Update predict using history.

Signed-off-by: ldwang <ftgreat@gmail.com>
FlagAI-Open · Oct 18, 2023 · 7b610ed · 7b610ed
1 parent 3875aa9
commit 7b610ed
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flagai/model/aquila2/modeling_aquila.py b/flagai/model/aquila2/modeling_aquila.py
@@ -1034,8 +1034,8 @@ def predict(self, text, tokenizer=None,
             probs = probs[1:]
 
         # Update history
-        history.insert(0, ('USER', text))
         history.insert(0, ('ASSISTANT', out))
+        history.insert(0, ('USER', text))
 
         return out 
 

diff --git a/flagai/model/aquila2_hf/predict.py b/flagai/model/aquila2_hf/predict.py
@@ -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))
+    history.insert(0, ('USER', text))
 
     return out