[PT FE] Fix timm model list according to newest version (#21376)
* [PT FE] Fix timm model list according to newest version * Remove xfail for hf model passed in latest nightly * Fix code for loading vit_h model * Restore torchvision
This commit is contained in:
parent
007b6fd82c
commit
3cae6e7aea
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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),)
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue