Spaces:

baulab
/

Erasing-Concepts-In-Diffusion

Runtime error

RohitGandikota commited on Mar 26, 2023

Commit

5a392b9

1 Parent(s): 2688a68

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -229,7 +229,7 @@ class Demo:
             n_steps=50,
             generator=generator
         )
         orig_image = images[0][0]
         torch.cuda.empty_cache()
@@ -243,7 +243,7 @@ class Demo:
                 n_steps=50,
                 generator=generator
             )
         edited_image = images[0][0]
         del finetuner

             n_steps=50,
             generator=generator
         )
+        images = self.diffuser.run_safety_checker(images=images, device='cuda')
         orig_image = images[0][0]
         torch.cuda.empty_cache()
                 n_steps=50,
                 generator=generator
             )
+        images = self.diffuser.run_safety_checker(images=images, device='cuda')
         edited_image = images[0][0]
         del finetuner