Update bianque_v2_app.py

scutcyr · Jun 6, 2023 · 7ea5c4a · 7ea5c4a
1 parent 1fb1156
commit 7ea5c4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bianque_v2_app.py b/bianque_v2_app.py
@@ -43,7 +43,7 @@
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # 指定模型名称或路径
-model_name_or_path = "scutcyr/BianQue-2.0"
+model_name_or_path = "scutcyr/BianQue-2"
 
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True)
 
@@ -155,4 +155,4 @@ def get_text():
     # Clear values from *all* all in-memory and on-disk data caches:
     # i.e. clear values from both square and cube
     st.session_state['generated'] = []
-    st.session_state['past'] = []
+    st.session_state['past'] = []