Spaces:
Running
on
Zero
Running
on
Zero
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
|