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:
Apophis3158 2025-12-08 07:09:44 +08:00 committed by GitHub
parent 56fa7dbe38
commit fb8bdb7590
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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):