Spaces:

HIT-TMG
/

dialogue-bart-large-chinese

Runtime error

YanshekWoo commited on Dec 13, 2022

Commit

e590807

•

1 Parent(s): fd3761b

Delete attention mask

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,8 +26,7 @@ def chat(history):
                         max_length=512)
     input_ids = outputs.input_ids
-    attention_mask = outputs.attention_mask
-    output_ids = model.generate(input_ids, attention_mask)[0]
     return tokenizer.decode(output_ids, skip_special_tokens=True)

                         max_length=512)
     input_ids = outputs.input_ids
+    output_ids = model.generate(input_ids)[0]
     return tokenizer.decode(output_ids, skip_special_tokens=True)