Rename Aquila2 70B names.

Signed-off-by: ldwang <[email protected]>
FlagAI-Open · Nov 29, 2023 · 1c03300 · 1c03300
1 parent 78a08ef
commit 1c03300
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
@@ -934,7 +934,7 @@ def predict(self, text, tokenizer=None,
 
         template_map = {"AquilaChat2-7B": "aquila-v1",
                         "AquilaChat2-34B": "aquila-legacy",
-                        "AquilaChat2-70B": "aquila-v2",
+                        "AquilaChat2-70B-Expr": "aquila-v2",
                         "AquilaChat2-7B-16K": "aquila",
                         "AquilaChat2-34B-16K": "aquila"}
         if not convo_template:

diff --git a/flagai/model/aquila2_hf/predict.py b/flagai/model/aquila2_hf/predict.py
@@ -374,7 +374,7 @@ def predict(model, text, tokenizer=None,
 
     template_map = {"AquilaChat2-7B": "aquila-v1",
                     "AquilaChat2-34B": "aquila-legacy",
-                    "AquilaChat2-70B": "aquila-v2",
+                    "AquilaChat2-70B-Expr": "aquila-v2",
                     "AquilaChat2-7B-16K": "aquila",
                     "AquilaChat2-34B-16K": "aquila"}
     if not convo_template: