diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 8b12ab6a0d..63e25fdf75 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -1252,7 +1252,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