diff --git a/nodes.py b/nodes.py index b2c9902..d9437f0 100644 --- a/nodes.py +++ b/nodes.py @@ -2761,7 +2761,7 @@ class WanVideoSampler: if fun_ref_image is not None: fun_ref_input = fun_ref_image.to(z) else: - fun_ref_input = torch.zeros_like(z, dtype=z.dtype)[:, 0].unsqueeze(1) + #fun_ref_input = torch.zeros_like(z, dtype=z.dtype)[:, 0].unsqueeze(1) fun_ref_input = None if control_lora: diff --git a/skyreels/nodes.py b/skyreels/nodes.py index d90afa2..91f37ac 100644 --- a/skyreels/nodes.py +++ b/skyreels/nodes.py @@ -12,6 +12,8 @@ from ..wanvideo.utils.scheduling_flow_match_lcm import FlowMatchLCMScheduler from ..nodes import optimized_scale from einops import rearrange +from ..enhance_a_video.globals import disable_enhance + import comfy.model_management as mm from comfy.utils import load_torch_file, ProgressBar, common_upscale from comfy.clip_vision import clip_preprocess, ClipVisionModel @@ -305,6 +307,7 @@ class WanVideoDiffusionForcingSampler: "end_percent": unianimate_poses["end_percent"] } + disable_enhance() #not sure if this can work, disabling for now to avoid errors if it's enabled by another sampler freqs = None transformer.rope_embedder.k = None