diff --git a/tests/model_hub_tests/torch_tests/hf_transformers_models b/tests/model_hub_tests/torch_tests/hf_transformers_models index ac1e1a8d510..b63e9eae171 100644 --- a/tests/model_hub_tests/torch_tests/hf_transformers_models +++ b/tests/model_hub_tests/torch_tests/hf_transformers_models @@ -30,7 +30,7 @@ beomi/exKcBERT-kowiki,exbert,skip,Load problem bert-base-uncased,bert bigscience/bloom-560m,bloom Bono93/basic-cnn-example,simplecnn,skip,Load problem -caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr,swin2sr,xfail,Unsupported op prim::Constant +caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr,swin2sr camembert-base,camembert CarlCochet/trajectory-transformer-ant-expert-v2,trajectory_transformer,xfail,Tracing problem CEBaB/bert-base-uncased.CEBaB.causalm.ambiance.2-class.exclusive.seed_42,bert_causalm,skip,Load problem diff --git a/tests/model_hub_tests/torch_tests/test_torchvision_models.py b/tests/model_hub_tests/torch_tests/test_torchvision_models.py index dcaffe21c3d..56ee127c3c7 100644 --- a/tests/model_hub_tests/torch_tests/test_torchvision_models.py +++ b/tests/model_hub_tests/torch_tests/test_torchvision_models.py @@ -78,6 +78,9 @@ class TestTorchHubConvertModel(TestTorchConvertModel): self.inputs = prepare_frames_for_raft(model_name, [frames[75], frames[125]], [frames[76], frames[126]]) + elif "vit_h_14" in model_name: + self.example = (torch.randn(1, 3, 518, 518),) + self.inputs = (torch.randn(1, 3, 518, 518),) else: self.example = (torch.randn(1, 3, 224, 224),) self.inputs = (torch.randn(1, 3, 224, 224),) diff --git a/tests/model_hub_tests/torch_tests/timm_models b/tests/model_hub_tests/torch_tests/timm_models index 25c414da4e7..46c06498d3c 100644 --- a/tests/model_hub_tests/torch_tests/timm_models +++ b/tests/model_hub_tests/torch_tests/timm_models @@ -133,6 +133,9 @@ efficientvit_b0.r224_in1k,None efficientvit_b1.r224_in1k,None efficientvit_b2.r224_in1k,None efficientvit_b3.r224_in1k,None +efficientvit_l1.r224_in1k,None +efficientvit_l2.r224_in1k,None +efficientvit_l3.r224_in1k,None efficientvit_m0.r224_in1k,None efficientvit_m1.r224_in1k,None efficientvit_m2.r224_in1k,None @@ -295,6 +298,7 @@ repvgg_b2g4.rvgg_in1k,None repvgg_b3.rvgg_in1k,None repvgg_b3g4.rvgg_in1k,None repvgg_d2se.rvgg_in1k,None +repvit_m0_9.dist_300e_in1k,None repvit_m1.dist_in1k,None repvit_m2.dist_in1k,None repvit_m3.dist_in1k,None @@ -461,26 +465,31 @@ vgg19.tv_in1k,None vgg19_bn.tv_in1k,None visformer_small.in1k,None vit_base_patch14_dinov2.lvd142m,None +vit_base_patch14_reg4_dinov2.lvd142m,None vit_base_patch16_224.augreg2_in21k_ft_in1k,None vit_base_patch16_224_miil.in21k,None vit_base_patch16_clip_224.datacompxl,None +vit_base_patch16_clip_quickgelu_224.metaclip_2pt5b,None vit_base_patch16_rpn_224.sw_in1k,None +vit_base_patch16_siglip_224.webli,None vit_base_patch32_224.augreg_in1k,None -vit_base_patch32_clip_224.laion2b,None +vit_base_patch32_clip_224.datacompxl,None +vit_base_patch32_clip_quickgelu_224.metaclip_2pt5b,None vit_base_patch8_224.augreg2_in21k_ft_in1k,None vit_base_r50_s16_224.orig_in21k,None vit_giant_patch14_clip_224.laion2b,None -vit_gigantic_patch16_224_ijepa.in22k,None +vit_giant_patch16_gap_224.in22k_ijepa,None vit_huge_patch14_224.mae,None -vit_huge_patch14_224_ijepa.in1k,None +vit_huge_patch14_clip_quickgelu_224.dfn5b,None +vit_huge_patch14_gap_224.in1k_ijepa,None vit_large_r50_s32_224.augreg_in21k,None -vit_medium_patch16_gap_240.sw_in12k,None vit_relpos_base_patch16_224.sw_in1k,None vit_relpos_base_patch16_clsgap_224.sw_in1k,None vit_relpos_base_patch32_plus_rpn_256.sw_in1k,None vit_relpos_medium_patch16_cls_224.sw_in1k,None vit_relpos_medium_patch16_rpn_224.sw_in1k,None vit_small_r26_s32_224.augreg_in21k,None +vit_so400m_patch14_siglip_224.webli,None vit_srelpos_medium_patch16_224.sw_in1k,None vit_tiny_r_s16_p8_224.augreg_in21k,None volo_d1_224.sail_in1k,None,xfail,Unsupported aten::col2im diff --git a/tests/model_hub_tests/torch_tests/torchvision_models b/tests/model_hub_tests/torch_tests/torchvision_models index a13a0f606c6..35e6805bd18 100644 --- a/tests/model_hub_tests/torch_tests/torchvision_models +++ b/tests/model_hub_tests/torch_tests/torchvision_models @@ -88,10 +88,10 @@ vgg16,none vgg16_bn,none vgg19,none vgg19_bn,none -vit_b_16,none,xfail,Tracing fails -vit_b_32,none,xfail,Tracing fails -vit_h_14,none,xfail,Tracing fails -vit_l_16,none,xfail,Tracing fails -vit_l_32,none,xfail,Tracing fails +vit_b_16,none +vit_b_32,none +vit_h_14,none +vit_l_16,none +vit_l_32,none wide_resnet101_2,none wide_resnet50_2,none