Create preprocessor_config.json

#1
by emobobas - opened
Files changed (1) hide show
  1. preprocessor_config.json +12 -0
preprocessor_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "processor_class": "AutoImageProcessor",
3
+ "pretrained_model_name": "ima806/deepfake_vs_real_image_detection",
4
+ "image_processing": {
5
+ "convert_to_rgb": true,
6
+ "return_tensors": "pt"
7
+ },
8
+ "data_collator": {
9
+ "type": "custom",
10
+ "description": "CustomDataCollator to stack image tensors and labels"
11
+ }
12
+ }