Spaces:
Configuration error
Configuration error
Commit
·
dfc0bf2
1
Parent(s):
247e28f
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-NoSleep-355m")
|
|
7 |
from happytransformer import GENSettings
|
8 |
|
9 |
def generate(text, length=100, penalty=3, temperature=0.8):
|
10 |
-
args_top_k = GENSettings(no_repeat_ngram_size=penalty, do_sample=True, top_k=80, temperature=temperature, max_length=length, early_stopping=False)
|
11 |
print("Prompt: " + text)
|
12 |
|
13 |
inputText = "[WP] " + text + " [RESPONSE] "
|
|
|
7 |
from happytransformer import GENSettings
|
8 |
|
9 |
def generate(text, length=100, penalty=3, temperature=0.8):
|
10 |
+
args_top_k = GENSettings(no_repeat_ngram_size=penalty, do_sample=True, top_k=80, top_p=1, temperature=temperature, max_length=length, early_stopping=False)
|
11 |
print("Prompt: " + text)
|
12 |
|
13 |
inputText = "[WP] " + text + " [RESPONSE] "
|