CreitinGameplays commited on
Commit
fab30bb
1 Parent(s): 435d814

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,10 +28,10 @@ def generate_text(user_prompt):
28
  num_beams=1,
29
  num_return_sequences=1,
30
  do_sample=True,
31
- top_k=50,
32
- top_p=0.95,
33
  temperature=0.2,
34
- repetition_penalty=1.155
35
  )
36
 
37
  # Decode the generated token sequence back to text
 
28
  num_beams=1,
29
  num_return_sequences=1,
30
  do_sample=True,
31
+ top_k=0,
32
+ top_p=1.0,
33
  temperature=0.2,
34
+ repetition_penalty=1.1
35
  )
36
 
37
  # Decode the generated token sequence back to text