model_config.yaml

channels: 4
depth: 18
head_dim: 64
in_dim: 128
mlp_ratio: 4.0
num_classes: 1001
num_heads: 12
patch_size: 4
vae: true
vae_scale_factor: 0.18215