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

update chatglm_llm.py

Browse Source
This commit is contained in:
imClumsyPanda 2023-04-22 21:20:18 +08:00
parent 5d88f7158a
commit 59aff96add
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/chatglm_llm.py
View File

@ -115,7 +115,8 @@ class ChatGLM(LLM):
prefix_encoder_file.close() prefix_encoder_file.close()
model_config.pre_seq_len = prefix_encoder_config['pre_seq_len'] model_config.pre_seq_len = prefix_encoder_config['pre_seq_len']
model_config.prefix_projection = prefix_encoder_config['prefix_projection'] model_config.prefix_projection = prefix_encoder_config['prefix_projection']
except Exception: except Exception as e:
print(e)
print("加载PrefixEncoder config.json失败") print("加载PrefixEncoder config.json失败")
if torch.cuda.is_available() and llm_device.lower().startswith("cuda"): if torch.cuda.is_available() and llm_device.lower().startswith("cuda"):