Unverified Commit 848f9fe6 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Core] pass revision in the loading_kwargs. (#7019)

* pass revision in the loading_kwarhs.

* remove revision from load_sub_model.
parent 8a692739
......@@ -436,7 +436,6 @@ def load_sub_model(
variant: str,
low_cpu_mem_usage: bool,
cached_folder: Union[str, os.PathLike],
revision: str = None,
):
"""Helper method to load the module `name` from `library_name` and `class_name`"""
# retrieve class candidates
......@@ -504,6 +503,7 @@ def load_sub_model(
loading_kwargs["offload_folder"] = offload_folder
loading_kwargs["offload_state_dict"] = offload_state_dict
loading_kwargs["variant"] = model_variants.pop(name, None)
if from_flax:
loading_kwargs["from_flax"] = True
......@@ -1280,7 +1280,6 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
variant=variant,
low_cpu_mem_usage=low_cpu_mem_usage,
cached_folder=cached_folder,
revision=revision,
)
logger.info(
f"Loaded {name} as {class_name} from `{name}` subfolder of {pretrained_model_name_or_path}."
......
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