anas-awadalla commited on
Commit
8a1b69a
·
1 Parent(s): 7b6d0c9

different examples

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -157,9 +157,9 @@ def generate(
157
  lang_x=input_ids,
158
  attention_mask=attention_mask,
159
  max_new_tokens=30,
160
- num_beams=5,
161
  do_sample=True,
162
- top_p=0.95,
163
  top_k=0,
164
  )
165
 
 
157
  lang_x=input_ids,
158
  attention_mask=attention_mask,
159
  max_new_tokens=30,
160
+ num_beams=3,
161
  do_sample=True,
162
+ temperature=0.7,
163
  top_k=0,
164
  )
165