diff --git a/main.py b/main.py index 68d4f29..753534f 100644 --- a/main.py +++ b/main.py @@ -383,8 +383,8 @@ class PromptQueue: with self.mutex: self.history.pop(id_to_delete, None) -async def run(server, address='', port=8188): - await asyncio.gather(server.start(address, port), server.publish_loop()) +async def run(server, address='', port=8188, verbose=True): + await asyncio.gather(server.start(address, port, verbose), server.publish_loop()) def hijack_progress(server): from tqdm.auto import tqdm @@ -410,6 +410,10 @@ if __name__ == "__main__": else: address = '127.0.0.1' + dont_print = False + if '--dont-print-server' in sys.argv: + dont_print = True + port = 8188 try: p_index = sys.argv.index('--port') @@ -419,9 +423,9 @@ if __name__ == "__main__": if os.name == "nt": try: - loop.run_until_complete(run(server, address=address, port=port)) + loop.run_until_complete(run(server, address=address, port=port, verbose=not dont_print)) except KeyboardInterrupt: pass else: - loop.run_until_complete(run(server, address=address, port=port)) + loop.run_until_complete(run(server, address=address, port=port, verbose=not dont_print)) diff --git a/notebooks/comfyui_colab.ipynb b/notebooks/comfyui_colab.ipynb index acfe4cf..8d26e8b 100644 --- a/notebooks/comfyui_colab.ipynb +++ b/notebooks/comfyui_colab.ipynb @@ -36,7 +36,8 @@ "!git clone https://github.com/comfyanonymous/ComfyUI\n", "%cd ComfyUI\n", "!pip install -r requirements.txt\n", - "!pip install xformers" + "!pip install xformers\n", + "!sed -i 's/v1-inference.yaml/v1-inference_fp16.yaml/g' webshit/index.html" ] }, { @@ -86,20 +87,23 @@ { "cell_type": "markdown", "source": [ - "### Run ComfyUI \n", - "use the **fp16** model configs for more speed\n", + "### Run ComfyUI with localtunnel\n", "\n", - "You should see the ui appear in an iframe. If you get a 403 error, it's your firefox settings or an extension that's messing things up.\n", + "If you have issues with the previous way, you can try this way. It will also work on non colab.\n", "\n", - "If you want to open it in another window use the second link not the first one.\n" + "use the **fp16** model configs for more speed\n", + "\n" ], "metadata": { - "id": "gggggggggg" + "id": "kkkkkkkkkkkkkk" } }, { "cell_type": "code", "source": [ + "!npm install -g localtunnel\n", + "\n", + "import subprocess\n", "import threading\n", "import time\n", "import socket\n", @@ -111,17 +115,17 @@ " if result == 0:\n", " break\n", " sock.close()\n", - " from google.colab import output\n", - " output.serve_kernel_port_as_iframe(port, height=1024)\n", - " print(\"to open it in a window you can open this link here:\")\n", - " output.serve_kernel_port_as_window(port)\n", + " p = subprocess.Popen([\"lt\", \"--port\", \"{}\".format(port)], stdout=subprocess.PIPE)\n", + " for line in p.stdout:\n", + " print(line.decode(), end='')\n", + "\n", "\n", "threading.Thread(target=iframe_thread, daemon=True, args=(8188,)).start()\n", "\n", - "!python main.py --highvram" + "!python main.py --highvram --dont-print-server" ], "metadata": { - "id": "hhhhhhhhhh" + "id": "jjjjjjjjjjjjj" }, "execution_count": null, "outputs": [] @@ -129,23 +133,20 @@ { "cell_type": "markdown", "source": [ - "### Run ComfyUI with localtunnel\n", + "### Run ComfyUI with colab iframe (in case localtunnel doesn't work)\n", + "use the **fp16** model configs for more speed\n", "\n", - "If you have issues with the previous way, you can try this way. It will also work on non colab.\n", + "You should see the ui appear in an iframe. If you get a 403 error, it's your firefox settings or an extension that's messing things up.\n", "\n", - "use the **fp16** model configs for more speed\n", - "\n" + "If you want to open it in another window use the link.\n" ], "metadata": { - "id": "kkkkkkkkkkkkkk" + "id": "gggggggggg" } }, { "cell_type": "code", "source": [ - "!npm install -g localtunnel\n", - "\n", - "import subprocess\n", "import threading\n", "import time\n", "import socket\n", @@ -157,17 +158,17 @@ " if result == 0:\n", " break\n", " sock.close()\n", - " p = subprocess.Popen([\"lt\", \"--port\", \"{}\".format(port)], stdout=subprocess.PIPE)\n", - " for line in p.stdout:\n", - " print(line.decode(), end='')\n", - "\n", + " from google.colab import output\n", + " output.serve_kernel_port_as_iframe(port, height=1024)\n", + " print(\"to open it in a window you can open this link here:\")\n", + " output.serve_kernel_port_as_window(port)\n", "\n", "threading.Thread(target=iframe_thread, daemon=True, args=(8188,)).start()\n", "\n", - "!python main.py --highvram" + "!python main.py --highvram --dont-print-server" ], "metadata": { - "id": "jjjjjjjjjjjjj" + "id": "hhhhhhhhhh" }, "execution_count": null, "outputs": [] diff --git a/server.py b/server.py index cc7d4a9..8e76b48 100644 --- a/server.py +++ b/server.py @@ -190,7 +190,7 @@ class PromptServer(): msg = await self.messages.get() await self.send(*msg) - async def start(self, address, port): + async def start(self, address, port, verbose=True): runner = web.AppRunner(self.app) await runner.setup() site = web.TCPSite(runner, address, port) @@ -198,5 +198,6 @@ class PromptServer(): if address == '': address = '0.0.0.0' - print("Starting server\n") - print("To see the GUI go to: http://{}:{}".format(address, port)) + if verbose: + print("Starting server\n") + print("To see the GUI go to: http://{}:{}".format(address, port))