File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
src/diffusers/pipelines/pag Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -952,9 +952,6 @@ def __call__(
952952 generator ,
953953 )
954954
955- num_warmup_steps = max (len (timesteps ) - num_inference_steps * self .scheduler .order , 0 )
956- self ._num_timesteps = len (timesteps )
957-
958955 if self .do_perturbed_attention_guidance :
959956 original_attn_proc = self .transformer .attn_processors
960957 self ._set_pag_attn_processor (
@@ -963,6 +960,8 @@ def __call__(
963960 )
964961
965962 # 6. Denoising loop
963+ num_warmup_steps = max (len (timesteps ) - num_inference_steps * self .scheduler .order , 0 )
964+ self ._num_timesteps = len (timesteps )
966965 with self .progress_bar (total = num_inference_steps ) as progress_bar :
967966 for i , t in enumerate (timesteps ):
968967 if self .interrupt :
You can’t perform that action at this time.
0 commit comments