Add auto_map for BrainOCR model code
Browse files- config.json +6 -1
config.json
CHANGED
|
@@ -76,5 +76,10 @@
|
|
| 76 |
"rms_norm_eps": 1e-05,
|
| 77 |
"spatial_merge_size": 2
|
| 78 |
},
|
| 79 |
-
"vocab_size": 120818
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 80 |
}
|
|
|
|
| 76 |
"rms_norm_eps": 1e-05,
|
| 77 |
"spatial_merge_size": 2
|
| 78 |
},
|
| 79 |
+
"vocab_size": 120818,
|
| 80 |
+
"auto_map": {
|
| 81 |
+
"AutoConfig": "configuration_brain_ocr.BrainOCRConfig",
|
| 82 |
+
"AutoImageProcessor": "image_processing_brain_ocr.BrainOCRImageProcessor",
|
| 83 |
+
"AutoProcessor": "processing_brain_ocr.BrainOCRProcessor"
|
| 84 |
+
}
|
| 85 |
}
|