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

Update nodes.py

Browse Source
This commit is contained in:
kijai
2025-03-26 00:14:59 +02:00
parent d67c96e4b5
commit b52e5b92bd
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+4 -3
View File
@@ -1008,7 +1008,6 @@ class WanVideoTextEncode:
encoder.model.to(offload_device)
mm.soft_empty_cache()
prompt_embeds_dict = {
"prompt_embeds": context,
"negative_prompt_embeds": context_null,
@@ -1995,7 +1994,9 @@ class WanVideoSampler:
if experimental_args is not None:
video_attention_split_steps = experimental_args.get("video_attention_split_steps", [])
if video_attention_split_steps:
transformer.video_attention_split_steps = [int(x.strip()) for x in video_attention_split_steps.split(",")]
transformer.video_attention_split_steps = [int(x.strip()) for x in video_attention_split_steps.split(",")]
else:
transformer.video_attention_split_steps = []
use_zero_init = experimental_args.get("use_zero_init", True)
use_cfg_zero_star = experimental_args.get("cfg_zero_star", False)
zero_star_steps = experimental_args.get("zero_star_steps", 0)
@@ -2055,7 +2056,7 @@ class WanVideoSampler:
noise_pred_uncond=noise_pred_uncond[0].to(intermediate_device)
noise_pred_text = noise_pred_cond
#https://github.com/WeichenFan/CFG-Zero-star/
if use_cfg_zero_star:
positive_flat = noise_pred_text.view(batch_size, -1)