| { | |
| "model_type": "treat-mmtb-2026", | |
| "task": 2, | |
| "mode": "cls", | |
| "cls_spec": { | |
| "img_size": 512, | |
| "threshold": 0.35, | |
| "patch_start": 5, | |
| "tower": { | |
| "model_type": "dinov3_vit", | |
| "hidden_size": 1024, | |
| "intermediate_size": 4096, | |
| "num_hidden_layers": 24, | |
| "num_attention_heads": 16, | |
| "num_register_tokens": 4, | |
| "patch_size": 16, | |
| "image_size": 512, | |
| "hidden_act": "gelu", | |
| "attention_dropout": 0.0, | |
| "drop_path_rate": 0.0, | |
| "initializer_range": 0.02, | |
| "layer_norm_eps": 1e-05, | |
| "layerscale_value": 1.0, | |
| "key_bias": false, | |
| "query_bias": true, | |
| "value_bias": true, | |
| "proj_bias": true, | |
| "mlp_bias": true, | |
| "num_channels": 3, | |
| "pos_embed_rescale": 2.0, | |
| "rope_theta": 100.0, | |
| "use_gated_mlp": false | |
| }, | |
| "pool": { | |
| "num_heads": 8 | |
| }, | |
| "head": { | |
| "vision_hidden_dim": 1024, | |
| "hidden_dim": 512, | |
| "num_classes": 1, | |
| "dropout": 0.1 | |
| } | |
| }, | |
| "architectures": [ | |
| "MMTBModel" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_mmtb.MMTBConfig", | |
| "AutoModel": "modeling_mmtb.MMTBModel" | |
| }, | |
| "transformers_version": "4.56.1" | |
| } | |