mirror of https://github.com/THUDM/ChatGLM2-6B
remove readline import from cli_demo.py
parent
1679b014c6
commit
0a5051f2fd
|
@ -2,7 +2,6 @@ import os
|
|||
import platform
|
||||
import signal
|
||||
from transformers import AutoTokenizer, AutoModel
|
||||
import readline
|
||||
|
||||
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True)
|
||||
model = AutoModel.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True).cuda()
|
||||
|
|
Loading…
Reference in New Issue