Spaces:

susunghong
/

Self-Attention-Guidance

Runtime error

susunghong commited on Feb 20, 2023

Commit

5875d0d

1 Parent(s): 6127de1

debug steps

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,9 +74,9 @@ def main():
         seed = random.randint(0, 100000) if randomize_seed else seed
         generator = torch.manual_seed(seed)
-        ori_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=0.0).images[0]
         generator = torch.manual_seed(seed)
-        sag_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=sag_scale).images[0]
         return [ori_image, sag_image, seed]
     def reset():

         seed = random.randint(0, 100000) if randomize_seed else seed
         generator = torch.manual_seed(seed)
+        ori_image = pipe(prompt, generator=generator, num_inference_steps=steps, guidance_scale=cfg_scale, sag_scale=0.0).images[0]
         generator = torch.manual_seed(seed)
+        sag_image = pipe(prompt, generator=generator, num_inference_steps=steps, guidance_scale=cfg_scale, sag_scale=sag_scale).images[0]
         return [ori_image, sag_image, seed]
     def reset():