susunghong commited on
Commit
5875d0d
1 Parent(s): 6127de1

debug steps

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,9 +74,9 @@ def main():
74
  seed = random.randint(0, 100000) if randomize_seed else seed
75
 
76
  generator = torch.manual_seed(seed)
77
- ori_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=0.0).images[0]
78
  generator = torch.manual_seed(seed)
79
- sag_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=sag_scale).images[0]
80
  return [ori_image, sag_image, seed]
81
 
82
  def reset():
 
74
  seed = random.randint(0, 100000) if randomize_seed else seed
75
 
76
  generator = torch.manual_seed(seed)
77
+ ori_image = pipe(prompt, generator=generator, num_inference_steps=steps, guidance_scale=cfg_scale, sag_scale=0.0).images[0]
78
  generator = torch.manual_seed(seed)
79
+ sag_image = pipe(prompt, generator=generator, num_inference_steps=steps, guidance_scale=cfg_scale, sag_scale=sag_scale).images[0]
80
  return [ori_image, sag_image, seed]
81
 
82
  def reset():