From 5254abb3ece1fe51480b626de6593460e65bdec9 Mon Sep 17 00:00:00 2001
From: ZhangErling <45256786+ZhangErling@users.noreply.github.com>
Date: Fri, 24 Mar 2023 15:22:06 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dapi=E6=9C=8D=E5=8A=A1?=
 =?UTF-8?q?=E5=99=A8=E5=90=AF=E5=8A=A8=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

第31行【API】会因为大小写原因找不到API
第34行的下划线【chatglm_6b】会导致模型加载错误
---
 api.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api.py b/api.py
index 14a2d57..10f70b6 100644
--- a/api.py
+++ b/api.py
@@ -28,8 +28,8 @@ async def create_item(request: Request):
 
 
 if __name__ == '__main__':
-    uvicorn.run('API:app', host='0.0.0.0', port=8000, workers=1)
+    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()