diff --git a/test/test_models.py b/test/test_models.py index 2629c3c4dc1..29b57c60cca 100644 --- a/test/test_models.py +++ b/test/test_models.py @@ -348,8 +348,8 @@ def _check_input_backprop(model, inputs): skipped_big_models = { "vit_h_14": {("Windows", "cpu"), ("Windows", "cuda")}, "regnet_y_128gf": {("Windows", "cpu"), ("Windows", "cuda")}, - "mvit_v1_b": {("Windows", "cuda")}, - "mvit_v2_s": {("Windows", "cuda")}, + "mvit_v1_b": {("Windows", "cuda"), ("Linux", "cuda")}, + "mvit_v2_s": {("Windows", "cuda"), ("Linux", "cuda")}, }