Skip some export tests on torch 2.7 (#38677)

* skip

* fix

* better check

* Update import_utils.py

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Co-authored-by: Cyril Vallez <cyril.vallez@gmail.com>
This commit is contained in:
Yih-Dar
2025-06-12 12:47:15 +02:00
committed by GitHub
parent 27459025b8
commit 89c46b648d
9 changed files with 30 additions and 7 deletions

View File

@@ -4569,10 +4569,7 @@ class ModelTesterMixin:
# Export model
exported_model = torch.export.export(
model,
args=(),
kwargs=inputs_dict,
strict=True,
model, args=(), kwargs=inputs_dict, strict=getattr(self, "test_torch_exportable_strictly", True)
)
# Run exported model and eager model