AI/ComfyUI-WanVideoWrapper
1
0
Fork 0
mirror of https://github.com/kijai/ComfyUI-WanVideoWrapper.git synced 2026-01-26 23:41:35 +03:00
Code 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

6
nodes_sampler.py
View File

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