Unverified Commit 5e8415a3 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Fix custom code loading in Automodel (#12571)

update
parent 051c8a1c
...@@ -147,14 +147,13 @@ class AutoModel(ConfigMixin): ...@@ -147,14 +147,13 @@ class AutoModel(ConfigMixin):
"force_download", "force_download",
"local_files_only", "local_files_only",
"proxies", "proxies",
"resume_download",
"revision", "revision",
"token", "token",
] ]
hub_kwargs = {name: kwargs.pop(name, None) for name in hub_kwargs_names} hub_kwargs = {name: kwargs.pop(name, None) for name in hub_kwargs_names}
# load_config_kwargs uses the same hub kwargs minus subfolder and resume_download # load_config_kwargs uses the same hub kwargs minus subfolder and resume_download
load_config_kwargs = {k: v for k, v in hub_kwargs.items() if k not in ["subfolder", "resume_download"]} load_config_kwargs = {k: v for k, v in hub_kwargs.items() if k not in ["subfolder"]}
library = None library = None
orig_class_name = None orig_class_name = None
...@@ -205,7 +204,6 @@ class AutoModel(ConfigMixin): ...@@ -205,7 +204,6 @@ class AutoModel(ConfigMixin):
module_file=module_file, module_file=module_file,
class_name=class_name, class_name=class_name,
**hub_kwargs, **hub_kwargs,
**kwargs,
) )
else: else:
from ..pipelines.pipeline_loading_utils import ALL_IMPORTABLE_CLASSES, get_class_obj_and_candidates from ..pipelines.pipeline_loading_utils import ALL_IMPORTABLE_CLASSES, get_class_obj_and_candidates
......
...@@ -307,7 +307,6 @@ class ModularPipelineBlocks(ConfigMixin, PushToHubMixin): ...@@ -307,7 +307,6 @@ class ModularPipelineBlocks(ConfigMixin, PushToHubMixin):
"local_files_only", "local_files_only",
"local_dir", "local_dir",
"proxies", "proxies",
"resume_download",
"revision", "revision",
"subfolder", "subfolder",
"token", "token",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment