Fix from_config (huggingface#23246)

fix
gojiteji · Jun 5, 2023 · c6fc02c · c6fc02c
1 parent 01a81b9
commit c6fc02c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/transformers/models/auto/auto_factory.py b/src/transformers/models/auto/auto_factory.py
@@ -407,8 +407,7 @@ def from_config(cls, config, **kwargs):
                 repo_id, class_ref = class_ref.split("--")
             else:
                 repo_id = config.name_or_path
-            module_file, class_name = class_ref.split(".")
-            model_class = get_class_from_dynamic_module(repo_id, module_file + ".py", class_name, **kwargs)
+            model_class = get_class_from_dynamic_module(class_ref, repo_id, **kwargs)
             return model_class._from_config(config, **kwargs)
         elif type(config) in cls._model_mapping.keys():
             model_class = _get_model_class(config, cls._model_mapping)