From 0f338857325928db270e13350f82d0e8a924f17f Mon Sep 17 00:00:00 2001 From: zerodegress Date: Fri, 24 Mar 2023 17:25:54 +0800 Subject: [PATCH] fixed case --- api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api.py b/api.py index f3c88b1..10f70b6 100644 --- a/api.py +++ b/api.py @@ -31,5 +31,5 @@ if __name__ == '__main__': uvicorn.run('api:app', host='0.0.0.0', port=8000, workers=1) tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True) -model = AutoModel.from_pretrained("THUDM/chatglm_6b", trust_remote_code=True).half().cuda() +model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda() model.eval()