Commit d8bbbce5 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add vit large model into list

parent f886d62d

imgutils/tagging/wd14.py

+1 −1
Original line number Diff line number Diff line
@@ -43,7 +43,7 @@ MODEL_NAMES = { 
    "ConvNext_v3": CONV_V3_MODEL_REPO,

    "ViT_v3": VIT_V3_MODEL_REPO,



    "Vit_Large": VIT_LARGE_MODEL_REPO,

    "ViT_Large": VIT_LARGE_MODEL_REPO,

}

_DEFAULT_MODEL_NAME = 'SwinV2_v3'

zoo/wd14/sync.py

+1 −0
Original line number Diff line number Diff line
@@ -47,6 +47,7 @@ _FC_NODE_PREFIXES_FOR_V3 = { 
    "SwinV2_v3": ('core_model', 'head', 'fc'),

    "ConvNext_v3": ('core_model', 'head', 'fc'),

    "ViT_v3": ('core_model', 'head'),

    "ViT_Large": ('core_model', 'head'),

}