BAAI
/

AquilaChat2-70B-Expr

Text Generation

Model card Files Files and versions

ldwang commited on Nov 29, 2023

Commit

589784e

·

1 Parent(s): e6d0e2d

Update predict.py

Files changed (1) hide show

predict.py +1 -0

predict.py CHANGED Viewed

@@ -347,6 +347,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"}
     if not convo_template:
         convo_template=template_map.get(model_name, "aquila-chat")

     template_map = {"AquilaChat2-7B": "aquila-v1",
                     "AquilaChat2-34B": "aquila-legacy",
                     "AquilaChat2-7B-16K": "aquila",
+                    "AquilaChat2-70B-Expr": "aquila-v2",
                     "AquilaChat2-34B-16K": "aquila"}
     if not convo_template:
         convo_template=template_map.get(model_name, "aquila-chat")