ford442 commited on
Commit
736b0ac
·
verified ·
1 Parent(s): 00651e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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