Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -98,8 +98,8 @@ with gr.Blocks() as demo:
|
|
| 98 |
inputs['input_ids'] = torch.cat([past_ids, inputs.input_ids], dim=1)
|
| 99 |
outputs = infer.model.generate(
|
| 100 |
**inputs, past_key_values=state.get('past_key_values', None),
|
| 101 |
-
return_dict_in_generate=True, do_sample=
|
| 102 |
-
repetition_penalty=
|
| 103 |
max_new_tokens=512,
|
| 104 |
)
|
| 105 |
state['past_key_values'] = outputs.past_key_values
|
|
|
|
| 98 |
inputs['input_ids'] = torch.cat([past_ids, inputs.input_ids], dim=1)
|
| 99 |
outputs = infer.model.generate(
|
| 100 |
**inputs, past_key_values=state.get('past_key_values', None),
|
| 101 |
+
return_dict_in_generate=True, do_sample=False,
|
| 102 |
+
repetition_penalty=1.05,
|
| 103 |
max_new_tokens=512,
|
| 104 |
)
|
| 105 |
state['past_key_values'] = outputs.past_key_values
|