github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add pull_model.py

pull/275/head
regmiao 2023-03-28 17:52:19 +08:00
parent 9addb875ca
commit 8798563db9
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pull_model.py Normal file
View File

@ -0,0 +1,2 @@
from transformers import AutoModel
model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()