Skip to content

Commit 5b06c6f

Browse files
authored
fix qwen image max rope index to 10000 (#139)
1 parent a460711 commit 5b06c6f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

diffsynth_engine/models/qwen_image/qwen_image_dit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ def __init__(
4848
super().__init__()
4949
self.theta = theta
5050
self.axes_dim = axes_dim
51-
pos_index = torch.arange(1024)
52-
neg_index = torch.arange(1024).flip(0) * -1 - 1
51+
pos_index = torch.arange(10000)
52+
neg_index = torch.arange(10000).flip(0) * -1 - 1
5353
self.pos_freqs = torch.cat(
5454
[
5555
self.rope_params(pos_index, self.axes_dim[0], self.theta),

0 commit comments

Comments
 (0)