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

fix: fix wrong file type for processor in `modular_model_converter.py`
This commit is contained in:
Tony Wu
2024-09-27 00:00:39 +02:00
committed by GitHub
parent e32521bf24
commit e1b150862e

View File

@@ -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",
}