diff --git a/bianque_v1_v2_app.py b/bianque_v1_v2_app.py index 3035407..e35f9f8 100644 --- 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'] = [] \ No newline at end of file + st.session_state['past'] = []