Skip to content

Commit 3546916

Browse files
authored
Update pipeline_lumina.py
1 parent 6b7362a commit 3546916

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/diffusers/pipelines/lumina/pipeline_lumina.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -751,9 +751,12 @@ def __call__(
751751
prompt_embeds=prompt_embeds,
752752
negative_prompt_embeds=negative_prompt_embeds,
753753
prompt_attention_mask=prompt_attention_mask,
754-
callback_on_step_end_tensor_inputs=callback_on_step_end_tensor_inputs,
755754
negative_prompt_attention_mask=negative_prompt_attention_mask,
755+
callback_on_step_end_tensor_inputs=callback_on_step_end_tensor_inputs,
756756
)
757+
758+
self._guidance_scale = guidance_scale
759+
757760
cross_attention_kwargs = {}
758761

759762
# 2. Define call parameters
@@ -815,6 +818,8 @@ def __call__(
815818
latents,
816819
)
817820

821+
self._num_timesteps = len(timesteps)
822+
818823
# 6. Denoising loop
819824
with self.progress_bar(total=num_inference_steps) as progress_bar:
820825
for i, t in enumerate(timesteps):

0 commit comments

Comments
 (0)