Skip to content

Commit 8de045d

Browse files
Fix Qwen3 crash
PiperOrigin-RevId: 835409126
1 parent 142b565 commit 8de045d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/MaxText/layers/qwen3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -708,7 +708,7 @@ def __call__(
708708
# Loop over the number of sub-layers that make up one repeating pattern.
709709
for i in range(cfg.inhomogeneous_layer_cycle_interval):
710710
layer = getattr(self, f"layer_{i}")
711-
x = layer(
711+
x, _ = layer(
712712
x,
713713
decoder_segment_ids,
714714
decoder_positions,

0 commit comments

Comments
 (0)