16 lines
604 B
Python
16 lines
604 B
Python
"""ComfyUI-FlashVSR - Real-time Video Super-Resolution Node"""
|
|
|
|
print("\n[FlashVSR] Loading nodes...")
|
|
|
|
from .AILab_FlashVSR import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
# Check for optional SageAttention optimization
|
|
try:
|
|
from sageattention import sageattn
|
|
print("[FlashVSR] ✓ SageAttention detected (~20-30% speedup enabled)")
|
|
except ImportError:
|
|
print("[FlashVSR] ○ SageAttention not installed (optional speedup available)")
|
|
|
|
print(f"[FlashVSR] ✓ Loaded {len(NODE_CLASS_MAPPINGS)} node(s)\n")
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] |