| { | |
| "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" | |
| } | |
| } |