forked from bedovyy/ComfyUI_NAIDGenerator
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path__init__.py
More file actions
26 lines (20 loc) · 888 Bytes
/
Copy path__init__.py
File metadata and controls
26 lines (20 loc) · 888 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import sys
import subprocess
import importlib
def ensure_package(package, install_package_name=None):
# Try to import the package
try:
importlib.import_module(package)
except ImportError:
print(f"Package {package} is not installed. Installing now...")
if "python_embeded" in sys.executable or "python_embedded" in sys.executable:
pip_install = [sys.executable, "-s", "-m", "pip", "install"]
else:
pip_install = [sys.executable, "-m", "pip", "install"]
subprocess.check_call(pip_install + [install_package_name or package])
# else:
# print(f"Package {package} is already installed.")
ensure_package("dotenv", "python-dotenv")
ensure_package("argon2", "argon2-cffi")
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']