Update predict.py

FlagAI-Open · Oct 20, 2023 · 636bf34 · 636bf34
1 parent 5423044
commit 636bf34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flagai/model/aquila2_hf/predict.py b/flagai/model/aquila2_hf/predict.py
@@ -344,7 +344,7 @@ def predict(model, text, tokenizer=None,
     template_map = {"AquilaChat2-7B": "aquila-v1",
                     "AquilaChat2-34B": "aquila-legacy",
                     "AquilaChat2-7B-16K": "aquila",
-                    "AquilaChat2-34B-16K": "aquila-v1"}
+                    "AquilaChat2-34B-16K": "aquila"}
     if not convo_template:
         convo_template=template_map.get(model_name, "aquila-chat")