28 lines
974 B
Python
28 lines
974 B
Python
import importlib
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
あ
|
|
SYMBOL = "🍌"
|
|
NODE_SURFIX = f"|cgem156"
|
|
ROOT_NAME = f"cgem156 {SYMBOL}/"
|
|
|
|
scripts = ["batch_condition", "dart", "lortnoc", "attention_couple", "cd_tuner", "lora_merger", "multiple_lora_loader", "custom_samplers", "custom_schedulers", "scale_crafter", "aesthetic_shadow"]
|
|
|
|
try:
|
|
import timm
|
|
except ImportError:
|
|
print("timm is not installed. Skipping load wd-tagger.")
|
|
else:
|
|
scripts.append("wd-tagger")
|
|
|
|
for script in scripts:
|
|
module = importlib.import_module(f"custom_nodes.cgem156-ComfyUI.scripts.{script}")
|
|
if hasattr(module, 'NODE_CLASS_MAPPINGS'):
|
|
NODE_CLASS_MAPPINGS.update(getattr(module, 'NODE_CLASS_MAPPINGS'))
|
|
if hasattr(module, 'NODE_DISPLAY_NAME_MAPPINGS'):
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(getattr(module, 'NODE_DISPLAY_NAME_MAPPINGS'))
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|