Spaces:

baulab
/

Erasing-Concepts-In-Diffusion

Runtime error

RohitGandikota commited on Mar 25, 2023

Commit

8479dbd

1 Parent(s): 4c06e55

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ class Demo:
         self.training = False
-        self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda:0').eval().half()
         with gr.Blocks() as demo:
             self.layout()

         self.training = False
+        self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda:1').eval().half()
         with gr.Blocks() as demo:
             self.layout()