Update app.py
Browse files
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=
|
| 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_"):
|