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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -320,7 +320,7 @@ def scheduler_swap_callback(pipeline, step_index, timestep, callback_kwargs):
320
  prompt_embeds = callback_kwargs["prompt_embeds"]
321
  prompt_embeds = prompt_embeds.chunk(2)[-1]
322
  # update guidance_scale and prompt_embeds
323
- pipeline._guidance_scale = new_scale
324
  callback_kwargs["prompt_embeds"] = prompt_embeds
325
  return callback_kwargs
326
 
 
320
  prompt_embeds = callback_kwargs["prompt_embeds"]
321
  prompt_embeds = prompt_embeds.chunk(2)[-1]
322
  # update guidance_scale and prompt_embeds
323
+ pipeline._guidance_scale = 0.0
324
  callback_kwargs["prompt_embeds"] = prompt_embeds
325
  return callback_kwargs
326