BAAI
/

AquilaChat2-34B-16K

Text Generation

Model card Files Files and versions Community

ldwang commited on Oct 20, 2023

Commit

b0e9c7f

•

1 Parent(s): 944391d

Update predict.py

Files changed (1) hide show

predict.py +1 -1

predict.py CHANGED Viewed

@@ -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"}
     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-34B-16K": "aquila"}
     if not convo_template:
         convo_template=template_map.get(model_name, "aquila-chat")