Having issues with loading ViT_16_384 weights

[Divs1159]

Hi
I tried the following code.

class ViTSmall16(nn.Module):
def init(self, n_classes, pretrained=False):

    super(ViTSmall16, self).__init__()
    
    self.model = timm.create_model("vit_small_patch16_384", pretrained)
    #self.model = timm.create_model("vit_large_patch16_224", pretrained)
    self.model.head = nn.Linear(self.model.head.in_features, n_classes) # Classification head

def forward(self, x):
    x = self.model(x)
    return x  

model = ViTSmall16(n_classes=2, pretrained=True)

But I get the error:
RuntimeError: Expected has Record("version") to be true, but got false.

How to resolve this? I didn't get any issues with vit_large_patch16_224, only with vit_small_patch16_384.
Please help

[rwightman]

@Divs1159 I see no issues with that model, you may have a corrupted weight file (partially downloaded, etc), you should find it in cache and delete

Otherwise, full version / environment info needed to try a closer reproduction.

[Divs1159]

Thank you.