| import sys |
| import subprocess |
| import locale |
| import threading |
| import os |
|
|
| def handle_stream(stream, prefix): |
| stream.reconfigure(encoding=locale.getpreferredencoding(), errors='replace') |
| for msg in stream: |
| if prefix == '[!]' and ('it/s]' in msg or 's/it]' in msg) and ('%|' in msg or 'it [' in msg): |
| if msg.startswith('100%'): |
| print('\r' + msg, end="", file=sys.stderr), |
| else: |
| print('\r' + msg[:-1], end="", file=sys.stderr), |
| else: |
| if prefix == '[!]': |
| print(prefix, msg, end="", file=sys.stderr) |
| else: |
| print(prefix, msg, end="") |
|
|
| def process_wrap(cmd_str, cwd_path, handler=None): |
| process = subprocess.Popen(cmd_str, cwd=cwd_path, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, bufsize=1) |
|
|
| if handler is None: |
| handler = handle_stream |
|
|
| stdout_thread = threading.Thread(target=handler, args=(process.stdout, "")) |
| stderr_thread = threading.Thread(target=handler, args=(process.stderr, "[!]")) |
|
|
| stdout_thread.start() |
| stderr_thread.start() |
|
|
| stdout_thread.join() |
| stderr_thread.join() |
|
|
| return process.wait() |
|
|
| assert process_wrap([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"], cwd_path=os.path.dirname(os.path.realpath(__file__))) == 0, "ERROR: Failed to install requirements.txt. Please install them manually, and restart ComfyUI." |
|
|
| nodep_packages = [ |
| "kornia>=0.6.9", |
| "xformers>=0.0.20", |
| ] |
|
|
| assert process_wrap([sys.executable, "-m", "pip", "install", "--no-deps", *nodep_packages], cwd_path=os.path.dirname(os.path.realpath(__file__))) == 0, "ERROR: Failed to install last set of packages. Please install them manually, and restart ComfyUI." |
