Spaces:
Configuration error
Configuration error
Commit
·
0b24949
1
Parent(s):
13b2bc5
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-NoSleep-1.5b")
|
|
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=
|
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=False, top_k=80, top_p=1, temperature=temperature, max_length=length, early_stopping=False)
|
11 |
print("Prompt: " + text)
|
12 |
|
13 |
inputText = "[WP] " + text + " [RESPONSE] "
|