diff --git a/llmstudio/cli.py b/llmstudio/cli.py index 2c6197ec..9d8a7d37 100644 --- 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() \ No newline at end of file