Skip to content

Commit b0c9b1f

Browse files
authored
Update src/diffusers/models/transformers/transformer_qwenimage.py
1 parent ca8cc6b commit b0c9b1f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/transformer_qwenimage.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def forward(self, timestep, hidden_states):
156156

157157

158158
class QwenEmbedRope(nn.Module):
159-
def __init__(self, theta: int, axes_dim: list[int], scale_rope=False):
159+
def __init__(self, theta: int, axes_dim: List[int], scale_rope=False):
160160
super().__init__()
161161
self.theta = theta
162162
self.axes_dim = axes_dim

0 commit comments

Comments
 (0)