mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-12-17 18:13:01 +08:00
Fix duplicate model_management initialization in hunyuan_video upsampler.py
Current behavior: comfy/ldm/hunyuan_video/upsampler.py imports model_management directly as top-level modules. Double initialization and log spamming. Fix: Changed imports to absolute imports (from comfy import ...) to align with the rest of the codebase and ensure the singleton instances are used correctly.
This commit is contained in:
parent
56fa7dbe38
commit
fb8bdb7590
@ -3,7 +3,7 @@ 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, model_patcher
|
from comfy import model_management, model_patcher
|
||||||
|
|
||||||
class SRResidualCausalBlock3D(nn.Module):
|
class SRResidualCausalBlock3D(nn.Module):
|
||||||
def __init__(self, channels: int):
|
def __init__(self, channels: int):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user