google-labs-jules[bot]
c6997fd9c2
- Implement node registration in `__init__.py` to fix "Node does not exist" error. - Implement automatic fallback from `inductor` to `cudagraphs` in `torch.compile` when Triton is missing (Windows fix). - Optimize VAE `InflatedCausalConv3d` to use 2D convolution path for spatial-only operations, improving speed. - Optimize VAE `ResnetBlock3D` to support flattened 4D execution path to reduce reshape overhead. - Update `VideoAutoencoderKL` to default `force_upcast=False` for FP16 inference. - Add `conv2d`/`conv3d` dequantization support to `GGUFTensor` for GGUF model compatibility.
28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
"""
|
|
ComfyUI-SeedVR2_VideoUpscaler
|
|
Official SeedVR2 integration for ComfyUI
|
|
"""
|
|
|
|
from .src.optimization.compatibility import ensure_triton_compat # noqa: F401
|
|
from .src.interfaces import comfy_entrypoint, SeedVR2Extension
|
|
from .src.interfaces.video_upscaler import SeedVR2VideoUpscaler
|
|
from .src.interfaces.dit_model_loader import SeedVR2LoadDiTModel
|
|
from .src.interfaces.vae_model_loader import SeedVR2LoadVAEModel
|
|
from .src.interfaces.torch_compile_settings import SeedVR2TorchCompileSettings
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SeedVR2VideoUpscaler": SeedVR2VideoUpscaler,
|
|
"SeedVR2LoadDiTModel": SeedVR2LoadDiTModel,
|
|
"SeedVR2LoadVAEModel": SeedVR2LoadVAEModel,
|
|
"SeedVR2TorchCompileSettings": SeedVR2TorchCompileSettings
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SeedVR2VideoUpscaler": "SeedVR2 Video Upscaler",
|
|
"SeedVR2LoadDiTModel": "SeedVR2 (Down)Load DiT Model",
|
|
"SeedVR2LoadVAEModel": "SeedVR2 (Down)Load VAE Model",
|
|
"SeedVR2TorchCompileSettings": "SeedVR2 Torch Compile Settings"
|
|
}
|
|
|
|
__all__ = ["comfy_entrypoint", "SeedVR2Extension", "NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|