From 0a342ad4d096268f6c29f6fa3324e1d1caf8f96c Mon Sep 17 00:00:00 2001
From: Khalil Meftah <khalil.meftah@huggingface.co>
Date: Fri, 24 Jul 2026 09:28:04 +0200
Subject: [PATCH] fix: exclude _is_pad mask from images

---
 scripts/overfit_vf_variant.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/overfit_vf_variant.py b/scripts/overfit_vf_variant.py
index 390d3c627..ade7d25b1 100644
--- a/scripts/overfit_vf_variant.py
+++ b/scripts/overfit_vf_variant.py
@@ -152,7 +152,7 @@ def _collate_processed(samples):
         }
 
     keys = {
-        *[key for key in samples[0] if key.startswith("observation.images.")],
+        *[key for key in samples[0] if key.startswith("observation.images.") and not key.endswith("_is_pad")],
         OBS_LANGUAGE_TOKENS,
         OBS_LANGUAGE_ATTENTION_MASK,
     }