Files
DumasNodes/dumas_h3_longvideos.py

33 lines
1019 B
Python

# Dumas Long Videos wrapper.
#
# The Long Videos implementation is vendored from:
# https://huggingface.co/Smite79/MiniMax-H3-Longvideos
#
# Keep this wrapper small so the upstream sampler can be refreshed without
# touching the other Dumas helper/reference/prompt nodes.
try:
from .dumas_h3_longvideos_upstream import H3LongVideos
except ImportError: # pragma: no cover - direct module import in local tests
from dumas_h3_longvideos_upstream import H3LongVideos
NODE_CLASS_MAPPINGS = {
"DumasH3LongVideos": H3LongVideos,
"H3LongVideos": H3LongVideos,
"H3LongVideosFL2VA": H3LongVideos,
"H3LongVideosV1": H3LongVideos,
"H3LongVideosREF2VA": H3LongVideos,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"DumasH3LongVideos": "Dumas H3 Long Videos",
"H3LongVideos": "H3 Long Videos",
"H3LongVideosFL2VA": "H3 Long Videos",
"H3LongVideosV1": "H3 Long Videos",
"H3LongVideosREF2VA": "H3 Long Videos",
}
__all__ = ["H3LongVideos", "NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]