File size: 309 Bytes
c9d6148 |
1 2 3 4 5 6 7 8 9 10 11 12 |
{
"processor_class": "AutoImageProcessor",
"pretrained_model_name": "umm-maybe/AI-image-detector",
"image_processing": {
"convert_to_rgb": true,
"return_tensors": "pt"
},
"data_collator": {
"type": "custom",
"description": "CustomDataCollator to stack image tensors and labels"
}
} |