Test: generate with torch.compile(model.forward) as a fast test (#34544)

2025-01-28 14:10:38 +00:00
parent f48ecd7608
commit ece8c42488
25 changed files with 105 additions and 53 deletions
--- a/tests/models/idefics/test_modeling_idefics.py
+++ b/tests/models/idefics/test_modeling_idefics.py
@@ -780,10 +780,6 @@ class IdeficsForVisionText2TextTest(IdeficsModelTest, GenerationTesterMixin, uni
    def test_custom_4d_attention_mask(self):
        pass

-    @unittest.skip(reason="IDEFICS cannot compile due to dynamic control flow when checking inputs")
-    def test_generate_compile_model_forward(self):
-        pass
-
    @unittest.skip(reason="We only test the model that takes in multiple images")
    def test_model(self):
        pass