WAR: Disabling ViT tests until exporting with py311 is fixed

More info:

huggingface/pytorch-image-models#1946

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

tests/cpp/test_compiled_modules.cpp

@@ -59,6 +59,8 @@ INSTANTIATE_TEST_SUITE_P(
         PathAndInput({"tests/modules/mobilenet_v2_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}}),
         PathAndInput({"tests/modules/efficientnet_b0_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}}),
         PathAndInput({"tests/modules/bert_base_uncased_traced.jit.pt", {{1, 14}, {1, 14}}, {at::kInt, at::kInt}}),
-        PathAndInput({"tests/modules/vit_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}})));
+// NOTE: ViT tests are disabled until Python 3.11 issue is resolved
+// https://github.com/huggingface/pytorch-image-models/issues/1946 PathAndInput({"tests/modules/vit_scripted.jit.pt",
+// {{1, 3, 224, 224}}, {at::kFloat}})));
 
 #endif

tests/cpp/test_modules_as_engines.cpp

@@ -30,5 +30,7 @@ INSTANTIATE_TEST_SUITE_P(
         PathAndInput({"tests/modules/resnet18_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}}),
         PathAndInput({"tests/modules/mobilenet_v2_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}}),
         PathAndInput({"tests/modules/efficientnet_b0_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}}),
-        PathAndInput({"tests/modules/vit_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}})));
+// NOTE: ViT tests are disabled until Python 3.11 issue is resolved
+// https://github.com/huggingface/pytorch-image-models/issues/1946 PathAndInput({"tests/modules/vit_scripted.jit.pt",
+// {{1, 3, 224, 224}}, {at::kFloat}})));
 #endif

tests/modules/hub.py

@@ -1,12 +1,13 @@
+import json
+import os
+
+import custom_models as cm
+import timm
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import torchvision.models as models
-import timm
-from transformers import BertModel, BertTokenizer, BertConfig
-import os
-import json
-import custom_models as cm
+from transformers import BertConfig, BertModel, BertTokenizer
 
 torch.hub._validate_not_a_forked_repo = lambda a, b, c: True
 
@@ -50,10 +51,11 @@
         "model": timm.create_model("efficientnet_b0", pretrained=True),
         "path": "script",
     },
-    "vit": {
-        "model": timm.create_model("vit_base_patch16_224", pretrained=True),
-        "path": "script",
-    },
+    # NOTE: Disabling ViT until support in 3.11 is fixed https://github.com/huggingface/pytorch-image-models/issues/1946
+    # "vit": {
+    #     "model": timm.create_model("vit_base_patch16_224", pretrained=True),
+    #     "path": "script",
+    # },
     "pooling": {"model": cm.Pool(), "path": "trace"},
     "module_fallback": {"model": cm.ModuleFallbackMain(), "path": "script"},
     "loop_fallback_eval": {"model": cm.LoopFallbackEval(), "path": "script"},