Hugging Face's logo

Goodis
/
SwarmComfyExtra

Model card Files
xet
 Community
SwarmComfyExtra / __init__.py
Goodis's picture
Goodis
Upload 13 files
efc1c32 verified
Raw History Blame Contribute Delete
837 Bytes
import traceback
NODE_CLASS_MAPPINGS = {}
# RemBg doesn't work on all python versions and OS's
try:
from . import SwarmRemBg
NODE_CLASS_MAPPINGS.update(SwarmRemBg.NODE_CLASS_MAPPINGS)
except ImportError:
print("Error: [Swarm] RemBg not available")
traceback.print_exc()
# This uses FFMPEG which doesn't install itself properly on Macs I guess?
try:
from . import SwarmSaveAnimationWS
NODE_CLASS_MAPPINGS.update(SwarmSaveAnimationWS.NODE_CLASS_MAPPINGS)
except ImportError:
print("Error: [Swarm] SaveAnimationWS not available")
traceback.print_exc()
# Yolo uses Ultralytics, which is cursed
try:
from . import SwarmYolo
NODE_CLASS_MAPPINGS.update(SwarmYolo.NODE_CLASS_MAPPINGS)
except ImportError:
print("Error: [Swarm] Yolo not available")
traceback.print_exc()