Do sample
Browse files
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=
|
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
|