chenjoya commited on
Commit
851054e
·
verified ·
1 Parent(s): d181208

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=do_sample,
102
- repetition_penalty=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