Skip to content

Commit a09ca7f

Browse files
committed
refactors: block __init__ no longer accept args. remove update_states from pipeline blocks, add update_states to modularpipeline, remove multi-block support for modular pipeline, remove offload support on modular pipeline
1 parent 8c02572 commit a09ca7f

File tree

3 files changed

+158
-766
lines changed

3 files changed

+158
-766
lines changed

src/diffusers/guider.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,7 @@ def set_guider(self, pipeline, guider_kwargs: Dict[str, Any]):
356356
self._guidance_rescale = guidance_rescale
357357
self._batch_size = batch_size
358358
if not hasattr(pipeline, "original_attn_proc") or pipeline.original_attn_proc is None:
359-
self.original_attn_proc = pipeline.unet.attn_processors
359+
pipeline.original_attn_proc = pipeline.unet.attn_processors
360360
self._set_pag_attn_processor(
361361
model=pipeline.unet if hasattr(pipeline, "unet") else pipeline.transformer,
362362
pag_applied_layers=self.pag_applied_layers,
@@ -366,11 +366,11 @@ def set_guider(self, pipeline, guider_kwargs: Dict[str, Any]):
366366
def reset_guider(self, pipeline):
367367
if (
368368
self.do_perturbed_attention_guidance
369-
and hasattr(self, "original_attn_proc")
370-
and self.original_attn_proc is not None
369+
and hasattr(pipeline, "original_attn_proc")
370+
and pipeline.original_attn_proc is not None
371371
):
372-
pipeline.unet.set_attn_processor(self.original_attn_proc)
373-
self.original_attn_proc = None
372+
pipeline.unet.set_attn_processor(pipeline.original_attn_proc)
373+
pipeline.original_attn_proc = None
374374

375375
def maybe_update_guider(self, pipeline, timestep):
376376
pass

0 commit comments

Comments
 (0)