Unverified Commit 7c54a7b3 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Fix Custom Code loading (#12378)

* update

* update

* update
parent 09e777a3
...@@ -194,7 +194,7 @@ class AutoModel(ConfigMixin): ...@@ -194,7 +194,7 @@ class AutoModel(ConfigMixin):
has_remote_code = "auto_map" in config and cls.__name__ in config["auto_map"] has_remote_code = "auto_map" in config and cls.__name__ in config["auto_map"]
trust_remote_code = resolve_trust_remote_code(trust_remote_code, pretrained_model_or_path, has_remote_code) trust_remote_code = resolve_trust_remote_code(trust_remote_code, pretrained_model_or_path, has_remote_code)
if not (has_remote_code and trust_remote_code): if not has_remote_code and trust_remote_code:
raise ValueError( raise ValueError(
"Selected model repository does not happear to have any custom code or does not have a valid `config.json` file." "Selected model repository does not happear to have any custom code or does not have a valid `config.json` file."
) )
......
...@@ -323,7 +323,7 @@ class ModularPipelineBlocks(ConfigMixin, PushToHubMixin): ...@@ -323,7 +323,7 @@ class ModularPipelineBlocks(ConfigMixin, PushToHubMixin):
trust_remote_code = resolve_trust_remote_code( trust_remote_code = resolve_trust_remote_code(
trust_remote_code, pretrained_model_name_or_path, has_remote_code trust_remote_code, pretrained_model_name_or_path, has_remote_code
) )
if not (has_remote_code and trust_remote_code): if not has_remote_code and trust_remote_code:
raise ValueError( raise ValueError(
"Selected model repository does not happear to have any custom code or does not have a valid `config.json` file." "Selected model repository does not happear to have any custom code or does not have a valid `config.json` file."
) )
......
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