RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-06 06:49:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update chatglm_llm.py

Browse Source
This commit is contained in:
littlepanda0716 2023-04-07 09:58:44 +08:00
parent befcdf6e32
commit dfe966ed41
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
chatglm_llm.py
View File

@ -15,9 +15,6 @@ model = (
.cuda() .cuda()
) )
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()
class ChatGLM(LLM): class ChatGLM(LLM):
max_token: int = 10000 max_token: int = 10000
temperature: float = 0.1 temperature: float = 0.1