Enable PaliGemma conversion tests

- PaliGemma conversion tests crashes because RemoveSDPACompositeZeroMaskPass
  replaces zero masks (used by ImageDecoder) with identity op which leads
  XNNPACK to access out-of-range area when it tries to get mask tensors
- Disable high-level function until RemoveSDPACompositeZeroMaskPass is removed
- Verified that it passes 100% now

PiperOrigin-RevId: 716432635

Author: ai-edge-bot

ai_edge_torch/generative/examples/paligemma/image_encoder.py

@@ -135,7 +135,9 @@ def get_image_encoder_config() -> cfg.ModelConfig:
       image_embedding=image_embedding_config,
       block_configs=block_config,
       final_norm_config=norm_config,
-      enable_hlfb=True,
+      # TODO: b/377051577 - Once RemoveSDPACompositeZeroMaskPass is removed,
+      # enable_hlfb can be set to True. See b/383865404#comment3 for details.
+      # enable_hlfb=True,
   )
   return config
 

ai_edge_torch/generative/test/test_model_conversion_large.py

@@ -232,7 +232,7 @@ def _test_paligemma_model(self, decoder_class, decoder_config, atol, rtol):
       ai_edge_torch.config.in_oss,
       reason="tests with custom ops are not supported in oss",
   )
-  def disabled_test_paligemma1(self):
+  def test_paligemma1(self):
     self._test_paligemma_model(
         decoder.Decoder, decoder.get_fake_decoder_config, atol=1e-3, rtol=1e-5
     )
@@ -241,7 +241,7 @@ def disabled_test_paligemma1(self):
       ai_edge_torch.config.in_oss,
       reason="tests with custom ops are not supported in oss",
   )
-  def disabled_test_paligemma2(self):
+  def test_paligemma2(self):
     self._test_paligemma_model(
         decoder2.Decoder2,
         decoder2.get_fake_decoder2_config,