Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -316,7 +316,7 @@ def scheduler_swap_callback(pipeline, step_index, timestep, callback_kwargs):
|
|
316 |
#pipe.scheduler.set_timesteps(num_inference_steps*.70)
|
317 |
# print(f"-- setting step {pipeline.num_timesteps * 0.9} --")
|
318 |
# pipeline.scheduler._step_index = pipeline.num_timesteps * 0.9
|
319 |
-
if step_index == int(pipeline.num_timesteps * self.config.cutoff_step_ratio)
|
320 |
prompt_embeds = callback_kwargs["prompt_embeds"]
|
321 |
prompt_embeds = prompt_embeds.chunk(2)[-1]
|
322 |
# update guidance_scale and prompt_embeds
|
|
|
316 |
#pipe.scheduler.set_timesteps(num_inference_steps*.70)
|
317 |
# print(f"-- setting step {pipeline.num_timesteps * 0.9} --")
|
318 |
# pipeline.scheduler._step_index = pipeline.num_timesteps * 0.9
|
319 |
+
if step_index == int(pipeline.num_timesteps * self.config.cutoff_step_ratio):
|
320 |
prompt_embeds = callback_kwargs["prompt_embeds"]
|
321 |
prompt_embeds = prompt_embeds.chunk(2)[-1]
|
322 |
# update guidance_scale and prompt_embeds
|