File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
src/diffusers/pipelines/lumina Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments