disable

Author: xadupre

_unittests/ut_torch_models/test_llm_phi2.py

@@ -25,6 +25,7 @@ def test_export_phi2_1(self):
         assert ep
 
     @ignore_warnings(UserWarning)
+    @requires_transformers("4.52")  # TODO
     def test_export_phi2_2_bypassed(self):
         data = get_phi2(num_hidden_layers=2)
         model, inputs, ds = data["model"], data["inputs"], data["dynamic_shapes"]
@@ -35,6 +36,10 @@ def test_export_phi2_2_bypassed(self):
             inputs = modificator(inputs)
             ep = torch.export.export(model, (), kwargs=inputs, dynamic_shapes=ds, strict=False)
             assert ep
+        with bypass_export_some_errors(patch_transformers=True) as modificator:
+            inputs = modificator(inputs)
+            ep = torch.export.export(model, (), kwargs=inputs, dynamic_shapes=ds, strict=False)
+            assert ep
 
 
 if __name__ == "__main__":