Update bianque_v1_v2_app.py

scutcyr · Jun 6, 2023 · 2cdaa04 · 2cdaa04
1 parent 7ea5c4a
commit 2cdaa04
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bianque_v1_v2_app.py b/bianque_v1_v2_app.py
@@ -45,7 +45,7 @@
 
 # 指定模型名称或路径
 bianque_v1_model_name_or_path = "scutcyr/BianQue-1.0"
-bianque_v2_model_name_or_path = "scutcyr/BianQue-2.0"
+bianque_v2_model_name_or_path = "scutcyr/BianQue-2"
 
 bianque_v1_tokenizer = T5Tokenizer.from_pretrained(bianque_v1_model_name_or_path)
 bianque_v2_tokenizer = AutoTokenizer.from_pretrained(bianque_v2_model_name_or_path, trust_remote_code=True)
@@ -237,4 +237,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'] = []