Comment on processor test

huggingface · Dec 5, 2024 · b74f736 · b74f736
1 parent cbfdc26
commit b74f736
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/models/pixtral/test_processor_pixtral.py b/tests/models/pixtral/test_processor_pixtral.py
@@ -276,6 +276,8 @@ def prepare_image_inputs(self, batch_size: Optional[int] = None):
         return [[super().prepare_image_inputs()]] * batch_size
 
     def test_batch_feature_mix_to(self):
+        # This test is here because BatchFeatureMix.to() was breaking the structure of some inputs
+        # so we ensure it doesn't regress
         processor = self.processor_class.from_pretrained(self.tmpdirname)
         prompt_string = "USER: [IMG][IMG]\nWhat's the difference between these two images? ASSISTANT:"