Hugging Face's logo

Spaces:
Heng666
/
tcml-chatglm-chat
Runtime error

App Files Community
Heng666 commited on
Commit
f9f1116
·
verified ·
1 Parent(s): 0f19430

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from threading import Thread
6
 
7
  # Loading the tokenizer and model from Hugging Face's model hub.
8
  # model_name_or_path = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
9
- model_name_or_path = "scutcyr/BianQue-2"
10
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path,trust_remote_code=True)
11
  # model = AutoModelForCausalLM.from_pretrained(model_name,trust_remote_code=True)
12
  model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True).half()
 
6
 
7
  # Loading the tokenizer and model from Hugging Face's model hub.
8
  # model_name_or_path = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
9
+ model_name_or_path = "scutcyr/BianQue-1.0"
10
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path,trust_remote_code=True)
11
  # model = AutoModelForCausalLM.from_pretrained(model_name,trust_remote_code=True)
12
  model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True).half()