Skip to content

Commit 70f40d6

Browse files
committed
fixed typo
1 parent c22ceeb commit 70f40d6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

MaxText/tests/multi_token_prediction_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def __call__(
139139
return self.mtp_block(
140140
main_hidden_state,
141141
self.shared_embedding,
142-
self.self.output_head,
142+
self.output_head,
143143
input_ids,
144144
target_ids,
145145
target_mask,

0 commit comments

Comments
 (0)