NickyNicky commited on
Commit
a0855f0
Β·
verified Β·
1 Parent(s): 9b7a7c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -73,10 +73,10 @@ def generate_text(prompt, max_length=2100):
73
  max_new_tokens=max_length
74
  generation_config = GenerationConfig(
75
  max_new_tokens=max_new_tokens,
76
- temperature=0.32,
77
  #top_p=0.9,
78
- top_k=50, # 45
79
- repetition_penalty=1.04, #1.1
80
  do_sample=True,
81
  )
82
  outputs = model.generate(generation_config=generation_config,
 
73
  max_new_tokens=max_length
74
  generation_config = GenerationConfig(
75
  max_new_tokens=max_new_tokens,
76
+ temperature=0.23,
77
  #top_p=0.9,
78
+ top_k=45, # 45
79
+ repetition_penalty=1., #1.1
80
  do_sample=True,
81
  )
82
  outputs = model.generate(generation_config=generation_config,