27 lines
858 B
Python
27 lines
858 B
Python
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.")
|
|
|
|
module_name = "diffusers"
|
|
|
|
ensure_package(module_name)
|
|
|
|
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|