Update modeling_aquila.py

FlagAI-Open · Oct 20, 2023 · 5423044 · 5423044
1 parent 13e7191
commit 5423044
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flagai/model/aquila2/modeling_aquila.py b/flagai/model/aquila2/modeling_aquila.py
@@ -935,7 +935,7 @@ def predict(self, 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")