Reduce requirements, use install script

This commit is contained in:
atmaranto
2024-03-07 03:59:46 -06:00
parent 9aaee0bc8c
commit 5fe795c13e
2 changed files with 13 additions and 13 deletions
+13 -1
View File
@@ -2,7 +2,6 @@ import sys
import os
from io import StringIO
import json
import traceback
@@ -11,6 +10,19 @@ from aiohttp import web
ext_dir = os.path.dirname(__file__)
sys.path.append(ext_dir)
try:
import black
except ImportError:
print("Unable to import requirements for ComfyUI-SaveAsScript.")
print("Installing...")
import importlib
spec = importlib.util.spec_from_file_location('impact_install', os.path.join(os.path.dirname(__file__), 'install.py'))
impact_install = importlib.util.module_from_spec(spec)
spec.loader.exec_module(impact_install)
print("Successfully installed. Hopefully, at least.")
# Prevent reimporting of custom nodes
os.environ["RUNNING_IN_COMFYUI"] = "TRUE"