Skip to content

Commit eaf6b47

Browse files
committed
fix
Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
1 parent a53e951 commit eaf6b47

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

nemo_automodel/components/datasets/vlm/collate_fns.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -384,8 +384,7 @@ def kimi_k25_vl_collate_fn(
384384
media_token_id = 163605 # Default for Kimi K2.5
385385

386386
pad_token_id = getattr(processor.tokenizer, "pad_token_id", 0) or 0
387-
rank = torch.distributed.get_rank() if torch.distributed.is_initialized() else 0
388-
387+
389388
# Process each sample individually
390389
all_expanded = []
391390
all_pixel_values = []

0 commit comments

Comments
 (0)