nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix default rope for v2 sampler

Browse Source
This commit is contained in:
kijai
2025-12-13 16:09:08 +02:00
parent 78e3e1857c
commit 0fa5383106
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes_sampler.py
+5 -1
View File
@@ -1056,6 +1056,8 @@ class WanVideoSampler:
freqs = None freqs = None
log.info(f"Rope function: {rope_function}")
riflex_freq_index = 0 if riflex_freq_index is None else riflex_freq_index riflex_freq_index = 0 if riflex_freq_index is None else riflex_freq_index
transformer.rope_embedder.k = None transformer.rope_embedder.k = None
transformer.rope_embedder.num_frames = None transformer.rope_embedder.num_frames = None
@@ -1063,7 +1065,7 @@ class WanVideoSampler:
if mocha_embeds is not None: if mocha_embeds is not None:
from .mocha.nodes import rope_params_mocha from .mocha.nodes import rope_params_mocha
log.info(f"Using Mocha RoPE") log.info("Using Mocha RoPE")
rope_function = 'mocha' rope_function = 'mocha'
freqs = torch.cat([ freqs = torch.cat([
@@ -3335,6 +3337,8 @@ class WanVideoSamplerv2(WanVideoSampler):
if extra_args is not None: if extra_args is not None:
args_dict.update(extra_args) args_dict.update(extra_args)
else:
args_dict["rope_function"] = "comfy"
return super().process(**args_dict) return super().process(**args_dict)