Infer bug for large model

[Vincent131499]

I am currently using 'visiontspp_large.ckpt' for inference, and I used the code from 'demo.ipynb', but the following error occurred.

==============================
size mismatch for blocks.9.norm2.weight: copying a param with shape torch.Size([1024]) from checkpoint, the shape in current model is torch.Size([768]).
size mismatch for blocks.9.norm2.bias: copying a param with shape torch.Size([1024]) from checkpoint, the shape in current model is torch.Size([768]).
size mismatch for blocks.9.mlp.fc1.weight: copying a param with shape torch.Size([4096, 1024]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
size mismatch for blocks.9.mlp.fc1.bias: copying a param with shape torch.Size([4096]) from checkpoint, the shape in current model is torch.Size([3072]).
size mismatch for blocks.9.mlp.fc2.weight: copying a param with shape torch.Size([1024, 4096]) from checkpoint, the shape in current model is torch.Size([768, 3072]).
size mismatch for blocks.9.mlp.fc2.bias: copying a param with shape torch.Size([1024]) from checkpoint, the shape in current model is torch.Size([768]).

[HALF111]

Hi, thank you for your interest in our work!
Sorry for the late reply due to our recent busy schedule! This issue has been fixed in the newly uploaded "demo.ipynb" file.
To use the correct model configuration, please make sure to set MODEL_SIZE = "base" or MODEL_SIZE = "large" in the corresponding cell, depending on which checkpoint you're loading.
You can refer to the screenshot below for an example of the correct setting.

Hope this can solve your problem~