Spaces:

baulab
/

Erasing-Concepts-In-Diffusion

Runtime error

RohitGandikota commited on Mar 25, 2023

Commit

16e9d06

1 Parent(s): 8479dbd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ class Demo:
     def __init__(self) -> None:
         self.training = False
         self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda:1').eval().half()
         with gr.Blocks() as demo:

     def __init__(self) -> None:
         self.training = False
+        print(f'Available GPUs: {torch.cuda.device_count()}')
         self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda:1').eval().half()
         with gr.Blocks() as demo: