disable enhance always with DF

This commit is contained in:
kijai
2025-04-25 15:47:25 +03:00
parent 33b8f1c082
commit 031d8ac817
2 changed files with 4 additions and 1 deletions
+1 -1
View File
@@ -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:
+3
View File
@@ -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