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