AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix model offload bug when key isn't present (#5030)

Browse Source 
* fix model offload bug when key isn't present

* make style
This commit is contained in:
Dhruv Nair
2023-09-14 14:32:06 +05:30
committed by GitHub
parent 19edca82f1
commit e2033d2dff
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/pipeline_utils.py
View File

@@ -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