BiliSakura's picture
Upload CROMA transformers checkpoints with model card metadata
930dfab verified
Raw
History Blame
716 Bytes
{
"architectures": [
"CromaModel"
],
"model_type": "croma",
"dtype": "float32",
"transformers_version": "5.0.0",
"hidden_size": 1024,
"num_hidden_layers": 24,
"num_attention_heads": 16,
"patch_size": 8,
"image_size": 120,
"sar_channels": 2,
"optical_channels": 12,
"modality": "both",
"hidden_dropout_prob": 0.0,
"layer_norm_eps": 1e-05,
"initializer_range": 0.02,
"num_patches": 225,
"auto_map": {
"AutoConfig": "modeling_croma.CromaConfig",
"AutoModel": "modeling_croma.CromaModel"
},
"custom_pipelines": {
"croma-feature-extraction": {
"impl": "pipeline_croma.CROMAImageFeatureExtractionPipeline",
"pt": [
"AutoModel"
]
}
}
}