BAAI
/

AquilaChat2-34B

Text Generation

Model card Files Files and versions Community

ldwang commited on Oct 18, 2023

Commit

5220fea

•

1 Parent(s): cde5d63

Upload predict.py

Files changed (1) hide show

predict.py +1 -1

predict.py CHANGED Viewed

@@ -436,7 +436,7 @@ def predict(model, text, tokenizer=None,
         probs = probs[1:]
     # Update history
-    history.insert(0, ('USER', text))
     history.insert(0, ('ASSISTANT', out))
     return out

         probs = probs[1:]
     # Update history
     history.insert(0, ('ASSISTANT', out))
+    history.insert(0, ('USER', text))
     return out