We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a53e951 commit eaf6b47Copy full SHA for eaf6b47
nemo_automodel/components/datasets/vlm/collate_fns.py
@@ -384,8 +384,7 @@ def kimi_k25_vl_collate_fn(
384
media_token_id = 163605 # Default for Kimi K2.5
385
386
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
-
+
389
# Process each sample individually
390
all_expanded = []
391
all_pixel_values = []
0 commit comments