Skip to content

Commit 30ef94a

Browse files
committed
Store semantic guidance vectors
1 parent bc51ea0 commit 30ef94a

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion_3.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -950,6 +950,10 @@ def __call__(
950950

951951
if edit_momentum is None:
952952
edit_momentum = torch.zeros_like(noise_guidance)
953+
954+
if self.sem_guidance is None:
955+
self.sem_guidance = torch.zeros((num_inference_steps + 1, *noise_pred_text.shape))
956+
953957
if enable_edit_guidance:
954958
concept_weights = torch.zeros(
955959
(len(noise_pred_edit_concepts), noise_guidance.shape[0]),

0 commit comments

Comments
 (0)