zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-05 18:05:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix LTXAV crash when sampling without an audio latent (#15132)

Browse Source
This commit is contained in:
Barish Ozbay
2026-07-28 23:24:08 -04:00
committed by GitHub
parent 42d2aa5543
commit e651b7bef5
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/ldm/lightricks/model.py
View File

@@ -671,9 +671,9 @@ def freqs_cis_matrix(freqs, pad_size, split_mode, num_attention_heads, out_dtype
cos_freq = torch.cat((cos_padding, cos_freq), dim=-1)
sin_freq = torch.cat((sin_padding, sin_freq), dim=-1)
B, T, _ = cos_freq.shape
cos_freq = cos_freq.reshape(B, T, num_attention_heads, -1)
sin_freq = sin_freq.reshape(B, T, num_attention_heads, -1)
B, T, half_HD = cos_freq.shape
cos_freq = cos_freq.reshape(B, T, num_attention_heads, half_HD // num_attention_heads)
sin_freq = sin_freq.reshape(B, T, num_attention_heads, half_HD // num_attention_heads)
rotation_matrix = torch.stack(
(cos_freq, -sin_freq, sin_freq, cos_freq), dim=-1
)