Skip to content

Commit bfcf14a

Browse files
committed
Minor update to ut
Signed-off-by: Chang Liu (Enterprise Products) <[email protected]>
1 parent 85e40b1 commit bfcf14a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unittest/_torch/multimodal/test_multimodal_runtime.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def create_mock_runtime(self, num_unseen_mm_tokens: int, num_mm_tokens: int,
130130
runtime = Mock(spec=MultimodalRuntimeData)
131131
runtime.num_unseen_mm_tokens = num_unseen_mm_tokens
132132
runtime.num_mm_tokens = num_mm_tokens
133-
runtime.mm_token_lengths = mm_token_lengths
133+
runtime.total_mm_tokens = sum(mm_token_lengths)
134134

135135
return runtime
136136

0 commit comments

Comments
 (0)