Ruurd commited on
Commit
da4880c
·
1 Parent(s): 7ee1641
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ with gr.Blocks() as demo:
74
  for token_id in current_model.generate(
75
  **inputs,
76
  max_new_tokens=256,
77
- do_sample=False,
78
  return_dict_in_generate=True,
79
  output_scores=False
80
  ).sequences[0][inputs['input_ids'].shape[-1]:]: # skip input tokens
 
74
  for token_id in current_model.generate(
75
  **inputs,
76
  max_new_tokens=256,
77
+ do_sample=True,
78
  return_dict_in_generate=True,
79
  output_scores=False
80
  ).sequences[0][inputs['input_ids'].shape[-1]:]: # skip input tokens