TF SAM memory reduction (huggingface#23732)

* Extremely small change to TF SAM dummies to reduce memory usage on build * remove debug breakpoint * Debug print statement to track array sizes * More debug shape printing * More debug shape printing * Now remove the debug shape printing * make fixup * make fixup
gojiteji · Jun 5, 2023 · 96a9896 · 96a9896
1 parent 583ea96
commit 96a9896
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/sam/modeling_tf_sam.py b/src/transformers/models/sam/modeling_tf_sam.py
@@ -1155,7 +1155,7 @@ def dummy_inputs(self) -> Dict[str, tf.Tensor]:
         """
         VISION_DUMMY_INPUTS = tf.random.uniform(
             shape=(
-                3,
+                1,
                 self.config.vision_config.num_channels,
                 self.config.vision_config.image_size,
                 self.config.vision_config.image_size,