DarwinAnim8or commited on
Commit
0b24949
·
1 Parent(s): 13b2bc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=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] "
 
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] "