Upload folder using huggingface_hub

api.py

@@ -50,7 +50,7 @@ async def create_item(request: Request):
 
 
 if __name__ == '__main__':
-    tokenizer = AutoTokenizer.from_pretrained("~/wufugang/damoxing_test/chatglm/model/chatglm-6b", trust_remote_code=True)
-    model = AutoModel.from_pretrained("~/wufugang/damoxing_test/chatglm/model/chatglm-6b", trust_remote_code=True).half().cuda()
+    tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True)
+    model = AutoModel.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True).half().cuda()
     model.eval()
     uvicorn.run(app, host='0.0.0.0', port=8000, workers=1)

cli_demo.py

@@ -4,8 +4,8 @@ import signal
 from transformers import AutoTokenizer, AutoModel
 import readline
 
-tokenizer = AutoTokenizer.from_pretrained("~/wufugang/damoxing_test/chatglm/model/chatglm-6b", trust_remote_code=True)
-model = AutoModel.from_pretrained("~/wufugang/damoxing_test/chatglm/model/chatglm-6b", trust_remote_code=True).half().cuda()
+tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True)
+model = AutoModel.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True).half().cuda()
 model = model.eval()
 
 os_name = platform.system()

web_demo.py

@@ -2,8 +2,8 @@ from transformers import AutoModel, AutoTokenizer
 import gradio as gr
 import mdtex2html
 
-tokenizer = AutoTokenizer.from_pretrained("~/wufugang/damoxing_test/chatglm/model/chatglm-6b", trust_remote_code=True)
-model = AutoModel.from_pretrained("~/wufugang/damoxing_test/chatglm/model/chatglm-6b", trust_remote_code=True).half().cuda()
+tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True)
+model = AutoModel.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True).half().cuda()
 model = model.eval()
 
 """Override Chatbot.postprocess"""