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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user