remove readline import from cli_demo.py

pull/357/head
Lyon 2023-07-22 22:24:38 +08:00 committed by GitHub
parent 1679b014c6
commit 0a5051f2fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions

View File

@ -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()