wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-28 23:30:16 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix import issue. (#11746)

Browse Source
This commit is contained in:
comfyanonymous 2026-01-08 14:23:59 -08:00 committed by GitHub
parent 0f11869d55
commit 1a20656448
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comfy/ldm/hunyuan_video/upsampler.py
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):
@ -103,13 +103,13 @@ UPSAMPLERS = {
class HunyuanVideo15SRModel(): class HunyuanVideo15SRModel():
def __init__(self, model_type, config): def __init__(self, model_type, config):
self.load_device = model_management.vae_device() self.load_device = comfy.model_management.vae_device()
offload_device = model_management.vae_offload_device() offload_device = comfy.model_management.vae_offload_device()
self.dtype = model_management.vae_dtype(self.load_device) self.dtype = comfy.model_management.vae_dtype(self.load_device)
self.model_class = UPSAMPLERS.get(model_type) self.model_class = UPSAMPLERS.get(model_type)
self.model = self.model_class(**config).eval() self.model = self.model_class(**config).eval()
self.patcher = model_patcher.ModelPatcher(self.model, load_device=self.load_device, offload_device=offload_device) self.patcher = comfy.model_patcher.ModelPatcher(self.model, load_device=self.load_device, offload_device=offload_device)
def load_sd(self, sd): def load_sd(self, sd):
return self.model.load_state_dict(sd, strict=True) return self.model.load_state_dict(sd, strict=True)
@ -118,5 +118,5 @@ class HunyuanVideo15SRModel():
return self.model.state_dict() return self.model.state_dict()
def resample_latent(self, latent): def resample_latent(self, latent):
model_management.load_model_gpu(self.patcher) comfy.model_management.load_model_gpu(self.patcher)
return self.model(latent.to(self.load_device)) return self.model(latent.to(self.load_device))