Loewolf commited on
Commit
433119f
1 Parent(s): 7dff0fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,13 +16,13 @@ def generate_text(prompt):
16
  attention_mask=attention_mask,
17
  max_length=max_length,
18
  min_length=4,
19
- num_beams=5,
20
  no_repeat_ngram_size=2,
21
  early_stopping=True,
22
- temperature=0.9,
23
  top_p=0.95,
24
- top_k=50,
25
- length_penalty=9.0,
26
  do_sample=True,
27
  eos_token_id=tokenizer.eos_token_id,
28
  pad_token_id=tokenizer.eos_token_id
 
16
  attention_mask=attention_mask,
17
  max_length=max_length,
18
  min_length=4,
19
+ num_beams=20,
20
  no_repeat_ngram_size=2,
21
  early_stopping=True,
22
+ temperature=0.6,
23
  top_p=0.95,
24
+ top_k=70,
25
+ length_penalty=2.0,
26
  do_sample=True,
27
  eos_token_id=tokenizer.eos_token_id,
28
  pad_token_id=tokenizer.eos_token_id