From 542304405033233715fd0d584fcb9e7594278b8d Mon Sep 17 00:00:00 2001
From: ldwang <ldwang@baai.ac.cn>
Date: Fri, 20 Oct 2023 14:14:06 +0800
Subject: [PATCH] Update modeling_aquila.py

---
 flagai/model/aquila2/modeling_aquila.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flagai/model/aquila2/modeling_aquila.py b/flagai/model/aquila2/modeling_aquila.py
index 6310f628..ea1083cc 100755
--- 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")