Update cli.py (#148)

TensorOpsAI · Oct 16, 2024 · 2dc0229 · 2dc0229
1 parent 3567855
commit 2dc0229
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/llmstudio/cli.py b/llmstudio/cli.py
@@ -1,5 +1,6 @@
 import os
 import signal
+import threading
 
 import click
 
@@ -25,11 +26,12 @@ def server(ui):
 
     print("Servers are running. Press CTRL+C to stop.")
 
+    stop_event = threading.Event()
     try:
-        signal.pause()
+        stop_event.wait()  # Wait indefinitely until the event is set
     except KeyboardInterrupt:
         print("Shutting down servers...")
 
 
 if __name__ == "__main__":
-    main()
+    main()