Fix import issue.

This commit is contained in:
comfyanonymous 2026-01-08 17:14:20 -05:00 committed by GitHub
parent 5943fbf457
commit a3caaab978
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -3,8 +3,8 @@ import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from comfy.ldm.modules.diffusionmodules.model import ResnetBlock, VideoConv3d from comfy.ldm.modules.diffusionmodules.model import ResnetBlock, VideoConv3d
from comfy.ldm.hunyuan_video.vae_refiner import RMS_norm from comfy.ldm.hunyuan_video.vae_refiner import RMS_norm
import model_management import comfy.model_management
import model_patcher import comfy.model_patcher
class SRResidualCausalBlock3D(nn.Module): class SRResidualCausalBlock3D(nn.Module):
def __init__(self, channels: int): def __init__(self, channels: int):