Spaces:

lsb
/

pareidolia-but-cute

Runtime error

lsb commited on Apr 1, 2024

Commit

68b0b4d

1 Parent(s): 6b394d7

use cuda maybe

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def mask_image_factory(mask_text="ASK FOR\nA SNACK", width=default_width, height
     return resized
 preferred_device = "cuda" if torch.cuda.is_available() else ("mps" if torch.backends.mps.is_available() else "cpu")
-preferred_device = "cpu"
 preferred_dtype = torch.float32
 controlnet = ControlNetModel.from_pretrained(

     return resized
 preferred_device = "cuda" if torch.cuda.is_available() else ("mps" if torch.backends.mps.is_available() else "cpu")
+# preferred_device = "cpu"
 preferred_dtype = torch.float32
 controlnet = ControlNetModel.from_pretrained(