Spaces:

shauray
/

Upside-down-Diffusion

Runtime error

shauray commited on Sep 20, 2023

Commit

3418ef3

•

1 Parent(s): bd27660

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     variants="fp32",
     use_safetensor=True,
 )
-pipe.to("cuda")
 @torch.no_grad()
 def call(

     variants="fp32",
     use_safetensor=True,
 )
+pipe.to(device)
 @torch.no_grad()
 def call(