from .vit_base_patch16 import model


model.embed_dim = 1024
model.depth = 24
model.num_heads = 16