DJF-on-arm commited on
Commit
30df1ec
·
verified ·
1 Parent(s): f4dfe1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -190,7 +190,7 @@ def fast_decode_step(token, e_seq, h, c, decoder_model, e_mask=None):
190
  c = tf.convert_to_tensor(c, dtype=tf.float32)
191
  return decoder_model([token, e_seq, e_mask, h, c], training=False)
192
 
193
- def reply(text, max_decode_len=MAX_LEN, rep_penalty=1.3, beam_width=13, length_penalty=0.7, temperature=0.7):
194
  text = text.strip()
195
  if MoE:
196
  if not text.startswith("<TASK_"):
 
190
  c = tf.convert_to_tensor(c, dtype=tf.float32)
191
  return decoder_model([token, e_seq, e_mask, h, c], training=False)
192
 
193
+ def reply(text, max_decode_len=MAX_LEN, rep_penalty=1.3, beam_width=7, length_penalty=0.7, temperature=0.7):
194
  text = text.strip()
195
  if MoE:
196
  if not text.startswith("<TASK_"):