physics-vit-standard / preprocessor_config.json
JessicaE's picture
Upload Physics ViT model
f8f1d04 verified
{
"do_normalize": false,
"do_rescale": false,
"do_resize": true,
"image_mean": null,
"image_std": null,
"resample": 2,
"rescale_factor": null,
"size": {
"height": 224,
"width": 224
},
"image_processor_type": "ViTImageProcessor",
"crop_pct": null,
"do_center_crop": false,
"processor_class": "ViTImageProcessor",
"preprocessing_source": "cerebras_modelzoo",
"preprocessing_url": "https://github.com/Cerebras/modelzoo/blob/5e81c965c68fd0a7ed9154bd0a7ed9154bd0ae26381f01218cd/src/cerebras/modelzoo/data/vision/transforms.py#L360",
"custom_preprocessing": {
"expand_to_square": true,
"background_color": [
128,
128,
128
],
"interpolation": "bilinear",
"antialias": true,
"transforms_sequence": [
"expand_to_square",
"resize",
"to_tensor"
],
"note": "This model requires Cerebras ModelZoo preprocessing pipeline. Standard HuggingFace ViTImageProcessor will NOT work correctly without the expand_to_square preprocessing step."
}
}