From 0fa53831064c1bc87c289cb00657c784e8204fa1 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sat, 13 Dec 2025 16:09:08 +0200 Subject: [PATCH] Fix default rope for v2 sampler --- nodes_sampler.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/nodes_sampler.py b/nodes_sampler.py index 5dde299..88fece9 100644 --- a/nodes_sampler.py +++ b/nodes_sampler.py @@ -1056,6 +1056,8 @@ class WanVideoSampler: freqs = None + log.info(f"Rope function: {rope_function}") + riflex_freq_index = 0 if riflex_freq_index is None else riflex_freq_index transformer.rope_embedder.k = None transformer.rope_embedder.num_frames = None @@ -1063,7 +1065,7 @@ class WanVideoSampler: if mocha_embeds is not None: from .mocha.nodes import rope_params_mocha - log.info(f"Using Mocha RoPE") + log.info("Using Mocha RoPE") rope_function = 'mocha' freqs = torch.cat([ @@ -3335,6 +3337,8 @@ class WanVideoSamplerv2(WanVideoSampler): if extra_args is not None: args_dict.update(extra_args) + else: + args_dict["rope_function"] = "comfy" return super().process(**args_dict)