From f3a14bd97fb26a1b61a35440a9d4c85e45aa526c Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 19 Feb 2024 09:35:33 +0530 Subject: [PATCH 1/2] pass revision in the loading_kwarhs. --- src/diffusers/pipelines/pipeline_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 18a4b5cb346b..b8b2d7dcc890 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -504,6 +504,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) + loading_kwargs["revision"] = revision if from_flax: loading_kwargs["from_flax"] = True From e54fdcc60bc917f85fe66e1875e450c5224a24c7 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 27 Feb 2024 07:12:25 +0530 Subject: [PATCH 2/2] remove revision from load_sub_model. --- src/diffusers/pipelines/pipeline_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index f4130d55c713..adb32a782b3e 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -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,7 +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) - loading_kwargs["revision"] = revision + if from_flax: loading_kwargs["from_flax"] = True @@ -1281,7 +1280,6 @@ def load_module(name, value): 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}."