diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 5ead6ef810..0eb684903e 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -1255,7 +1255,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): self._all_hooks = [] hook = None for model_str in self.model_cpu_offload_seq.split("->"): - model = all_model_components.pop(model_str) + model = all_model_components.pop(model_str, None) if not isinstance(model, torch.nn.Module): continue