298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update colab notebook to use an iframe for GUI.

Browse Source
This commit is contained in:
comfyanonymous 2023-01-27 21:48:48 -05:00
parent 9fa1827906
commit 266db2066d
1 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
notebooks/comfyui_colab.ipynb
View File

@ -63,7 +63,7 @@
{
"cell_type": "markdown",
"source": [
"Launch a http server to see the output pics:"
"Optional: Launch a http server to see the output pics (you can also download them by browsing to the output directory with colab):"
],
"metadata": {
"id": "eeeeeeeeee"
@ -94,10 +94,22 @@
{
"cell_type": "code",
"source": [
"from google.colab import output\n",
"print(\"open this next link instead once the software prints: To see the GUI go to\")\n",
"output.serve_kernel_port_as_window(8188)\n",
"!sed -i 's/127.0.0.1/0.0.0.0/g' main.py\n",
"import threading\n",
"import time\n",
"import socket\n",
"def iframe_thread(port):\n",
" while True:\n",
" time.sleep(0.5)\n",
" sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)\n",
" result = sock.connect_ex(('127.0.0.1', port))\n",
" if result == 0:\n",
" break\n",
" sock.close()\n",
" from google.colab import output\n",
" x = output.serve_kernel_port_as_window(port)\n",
" x = output.serve_kernel_port_as_iframe(port, height=1024)\n",
"threading.Thread(target=iframe_thread, daemon=True, args=(8188,)).start()\n",
"\n",
"!python main.py"
],
"metadata": {