sd/prompt_helper/model/sw_jax_cv_config.json

{
    "image_size": 448,
    "model_name": "vit_base",
    "model_args": {
        "patch_size": 16,
        "num_classes": 10861,
        "num_layers": 12,
        "embed_dim": 768,
        "mlp_dim": 3072,
        "num_heads": 12,
        "drop_path_rate": 0.1,
        "layer_norm_eps": 1e-05
    }
}