File size: 1,024 Bytes
7cc9dda | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | """
SeedVR2 ComfyUI Nodes
Central registry for all SeedVR2 nodes
"""
from comfy_api.latest import ComfyExtension, io
from .video_upscaler import SeedVR2VideoUpscaler
from .dit_model_loader import SeedVR2LoadDiTModel
from .vae_model_loader import SeedVR2LoadVAEModel
from .torch_compile_settings import SeedVR2TorchCompileSettings
class SeedVR2Extension(ComfyExtension):
"""SeedVR2 ComfyUI Extension"""
async def get_node_list(self) -> list[type[io.ComfyNode]]:
"""Return list of all SeedVR2 nodes"""
return [
SeedVR2VideoUpscaler,
SeedVR2LoadDiTModel,
SeedVR2LoadVAEModel,
SeedVR2TorchCompileSettings,
]
async def comfy_entrypoint() -> ComfyExtension:
"""ComfyUI V3 entry point"""
return SeedVR2Extension()
__all__ = [
'SeedVR2VideoUpscaler',
'SeedVR2LoadDiTModel',
'SeedVR2LoadVAEModel',
'SeedVR2TorchCompileSettings',
'SeedVR2Extension',
'comfy_entrypoint',
] |