We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0bc08eb commit dca8ec4Copy full SHA for dca8ec4
_unittests/ut_torch_export_patches/test_patch_torch.py
@@ -6,6 +6,7 @@
6
ExtTestCase,
7
requires_torch,
8
requires_transformers,
9
+ has_transformers,
10
has_torch,
11
)
12
from onnx_diagnostic.helpers.cache_helper import CacheKeyValue, make_dynamic_cache
@@ -361,6 +362,8 @@ def test_export_with_patch_tiny_llm_dim_meta(self):
361
362
self.assertEqualArrayAny(expected, got)
363
364
with self.subTest(input="no01", backed_size_oblivious=True):
365
+ if not has_transformers("4.55"):
366
+ raise unittest.SkipIf("test not working with transformers<4.55")
367
with (
368
torch.fx.experimental._config.patch(backed_size_oblivious=True),
369
torch_export_patches(patch_transformers=True),
0 commit comments