Fix support for image processors modifications in modular (#34866)

* add fix and examples * fix camel case naming
2024-11-22 18:14:24 -05:00
parent 54be2d7ae8
commit 3a8eb74668
3 changed files with 297 additions and 1 deletions
--- a/utils/modular_model_converter.py
+++ b/utils/modular_model_converter.py
@@ -1192,7 +1192,7 @@ class ModularFileMapper(ModuleMapper):
        self.visited_modules = {}
        self.renamers = {}
        for file, module in self.model_specific_modules.items():
-            file_model_name = re.search(r"models\.\w*?\.\w*?_(\S*)", file).groups()[0]
+            file_model_name = file.split(".")[-2]
            renamer = ReplaceNameTransformer(
                file_model_name, self.model_name, self.given_old_name, self.given_new_name
            )