Fix modular model converter unable to generate Processor classes (#33737

) fix: fix wrong file type for processor in `modular_model_converter.py`
huggingface · Sep 26, 2024 · e1b1508 · e1b1508
1 parent e32521b
commit e1b1508
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/modular_model_converter.py b/utils/modular_model_converter.py
@@ -502,7 +502,7 @@ def replace_call_to_super(class_finder: ClassFinder, updated_node: cst.ClassDef,
 TYPE_TO_FILE_TYPE = {
     "Config": "configuration",
     "Tokenizer": "tokenization",
-    "Processor": "processor",
+    "Processor": "processing",
     "ImageProcessor": "image_processing",
     "FeatureExtractor": "feature_extractor",
 }