if@impactframes
d4113f3c6d
- Add HunyuanVideoFoleyModelLoader with FP8 weight-only quantization (fp8_e4m3fn, fp8_e5m2) - Add HunyuanVideoFoleyDependenciesLoader for modular component loading - Add HunyuanVideoFoleyTorchCompile for ~30% inference speedup - Add HunyuanVideoFoleyGeneratorAdvanced that accepts pre-loaded models - Keep original HunyuanVideoFoley node fully functional for backward compatibility - Fix model path to use ComfyUI's foley/hunyuanvideo-foley-xxl directory structure - Support both standalone and pipeline-based workflows Co-Authored-By: if@impactframes <if@impactframes.ai>
15 lines
597 B
Python
15 lines
597 B
Python
import os
|
|
import sys
|
|
from loguru import logger
|
|
|
|
# Add the current directory to Python path to import hunyuanvideo_foley modules
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
if current_dir not in sys.path:
|
|
sys.path.insert(0, current_dir)
|
|
|
|
# Import the individual nodes (with FP8 quantization and torch.compile support)
|
|
logger.info("Loading HunyuanVideo-Foley nodes with FP8 quantization and torch.compile support")
|
|
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
# Export the mappings
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |