This commit is contained in:
hiyouga 2023-11-14 15:27:05 +08:00
parent 2d42be32c1
commit 3743b7420b
1 changed files with 7 additions and 1 deletions

View File

@ -1,6 +1,12 @@
import readline
from llmtuner import ChatModel from llmtuner import ChatModel
try:
import platform
if platform.system() != "Windows":
import readline
except ImportError:
print("Install `readline` for a better experience.")
def main(): def main():
chat_model = ChatModel() chat_model = ChatModel()