From 0a5051f2fdc9434db2afc077c09548aca5bdb5e7 Mon Sep 17 00:00:00 2001 From: Lyon <88232613+pikasTech@users.noreply.github.com> Date: Sat, 22 Jul 2023 22:24:38 +0800 Subject: [PATCH] remove readline import from cli_demo.py --- cli_demo.py | 1 - 1 file changed, 1 deletion(-) diff --git a/cli_demo.py b/cli_demo.py index 4aa35d3..f048084 100644 --- a/cli_demo.py +++ b/cli_demo.py @@ -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()