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

Fix rf-inversion shift

Browse Source
This commit is contained in:
kijai
2025-01-06 15:15:14 +02:00
parent e8e0ac4e56
commit 378b4a264d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes_rf_inversion.py
+3 -3
View File
@@ -121,7 +121,7 @@ class HyVideoInverseSampler:
freqs_cos, freqs_sin = get_rotary_pos_embed(transformer, latent_num_frames, height, width)
pipeline.scheduler.shift = flow_shift
pipeline.scheduler.flow_shift = flow_shift
if model["block_swap_args"] is not None:
for name, param in transformer.named_parameters():
@@ -336,7 +336,7 @@ class HyVideoReSampler:
freqs_cos, freqs_sin = get_rotary_pos_embed(transformer, latent_num_frames, height, width)
pipeline.scheduler.shift = flow_shift
pipeline.scheduler.flow_shift = flow_shift
if model["block_swap_args"] is not None:
for name, param in transformer.named_parameters():
@@ -535,7 +535,7 @@ class HyVideoPromptMixSampler:
latent_video_length = (num_frames - 1) // 4 + 1
freqs_cos, freqs_sin = get_rotary_pos_embed(transformer, latent_video_length, height, width)
pipeline.scheduler.shift = flow_shift
pipeline.scheduler.flow_shift = flow_shift
if model["block_swap_args"] is not None:
for name, param in transformer.named_parameters():