fix #1489

2023-11-14 15:27:05 +08:00 · 2023-11-14 15:27:05 +08:00 · 3743b7420b
parent 2d42be32c1
commit 3743b7420b
1 changed files with 7 additions and 1 deletions
--- a/src/cli_demo.py
+++ b/src/cli_demo.py
@ -1,6 +1,12 @@
-import readline
 from llmtuner import ChatModel

+try:
+    import platform
+    if platform.system() != "Windows":
+        import readline
+except ImportError:
+    print("Install `readline` for a better experience.")
+

 def main():
    chat_model = ChatModel()