diff --git a/src/diffusers/models/transformers/transformer_qwenimage.py b/src/diffusers/models/transformers/transformer_qwenimage.py index 846add8906ac..1922c17f0e3d 100644 --- a/src/diffusers/models/transformers/transformer_qwenimage.py +++ b/src/diffusers/models/transformers/transformer_qwenimage.py @@ -219,7 +219,7 @@ def forward(self, video_fhw, txt_seq_lens, device): self.rope_cache[rope_key] = self._compute_video_freqs(frame, height, width, idx) video_freq = self.rope_cache[rope_key] else: - video_freq = self._compute_video_freqs(frame, height, width, idx) + video_freq = self._compute_video_freqs.__wrapped__(self, frame, height, width, idx) video_freq = video_freq.to(device) vid_freqs.append(video_freq)