wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-27 17:37:39 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

fix: cross-attention AdaLN scale, shift, sigma parameters calculation

Browse Source
This commit is contained in:
izorin 2026-05-25 10:26:58 +00:00
parent 0077d78cbf
commit b496d401d7
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/ldm/lightricks/av_model.py
View File

@ -767,25 +767,25 @@ class LTXAVModel(LTXVModel):
# Cross-attention timesteps - compress these too # Cross-attention timesteps - compress these too
av_ca_audio_scale_shift_timestep, _ = self.av_ca_audio_scale_shift_adaln_single( av_ca_audio_scale_shift_timestep, _ = self.av_ca_audio_scale_shift_adaln_single(
timestep.max().expand_as(a_timestep_flat), a_timestep_flat,
{"resolution": None, "aspect_ratio": None}, {"resolution": None, "aspect_ratio": None},
batch_size=batch_size, batch_size=batch_size,
hidden_dtype=hidden_dtype, hidden_dtype=hidden_dtype,
) )
av_ca_video_scale_shift_timestep, _ = self.av_ca_video_scale_shift_adaln_single( av_ca_video_scale_shift_timestep, _ = self.av_ca_video_scale_shift_adaln_single(
a_timestep.max().expand_as(timestep_flat), timestep_flat,
{"resolution": None, "aspect_ratio": None}, {"resolution": None, "aspect_ratio": None},
batch_size=batch_size, batch_size=batch_size,
hidden_dtype=hidden_dtype, hidden_dtype=hidden_dtype,
) )
av_ca_a2v_gate_noise_timestep, _ = self.av_ca_a2v_gate_adaln_single( av_ca_a2v_gate_noise_timestep, _ = self.av_ca_a2v_gate_adaln_single(
a_timestep.max().expand_as(timestep_flat) * av_ca_factor, a_timestep_scaled.max().expand_as(timestep_flat) * av_ca_factor,
{"resolution": None, "aspect_ratio": None}, {"resolution": None, "aspect_ratio": None},
batch_size=batch_size, batch_size=batch_size,
hidden_dtype=hidden_dtype, hidden_dtype=hidden_dtype,
) )
av_ca_v2a_gate_noise_timestep, _ = self.av_ca_v2a_gate_adaln_single( av_ca_v2a_gate_noise_timestep, _ = self.av_ca_v2a_gate_adaln_single(
timestep.max().expand_as(a_timestep_flat) * av_ca_factor, timestep_scaled.max().expand_as(a_timestep_flat) * av_ca_factor,
{"resolution": None, "aspect_ratio": None}, {"resolution": None, "aspect_ratio": None},
batch_size=batch_size, batch_size=batch_size,
hidden_dtype=hidden_dtype, hidden_dtype=hidden_dtype,