Skip to content

Commit 35efa06

Browse files
committed
formatting
1 parent 270c63f commit 35efa06

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/models/transformers/transformer_qwenimage.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,9 @@ def forward(
304304
return vid_freqs, txt_freqs
305305

306306
@functools.lru_cache(maxsize=128)
307-
def _compute_video_freqs(self, frame: int, height: int, width: int, idx: int = 0, device: torch.device = None) -> torch.Tensor:
307+
def _compute_video_freqs(
308+
self, frame: int, height: int, width: int, idx: int = 0, device: torch.device = None
309+
) -> torch.Tensor:
308310
seq_lens = frame * height * width
309311
pos_freqs = self.pos_freqs.to(device) if device is not None else self.pos_freqs
310312
neg_freqs = self.neg_freqs.to(device) if device is not None else self.neg_freqs

0 commit comments

Comments
 (0)