Hugging Face's logo

Spaces:
KienPongPoh
/
pareidolia-but-cute
Sleeping

App Files Community
lsb commited on
Commit
68b0b4d
•
1 Parent(s): 6b394d7

use cuda maybe

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def mask_image_factory(mask_text="ASK FOR\nA SNACK", width=default_width, height
44
  return resized
45
 
46
  preferred_device = "cuda" if torch.cuda.is_available() else ("mps" if torch.backends.mps.is_available() else "cpu")
47
- preferred_device = "cpu"
48
  preferred_dtype = torch.float32
49
 
50
  controlnet = ControlNetModel.from_pretrained(
 
44
  return resized
45
 
46
  preferred_device = "cuda" if torch.cuda.is_available() else ("mps" if torch.backends.mps.is_available() else "cpu")
47
+ # preferred_device = "cpu"
48
  preferred_dtype = torch.float32
49
 
50
  controlnet = ControlNetModel.from_pretrained(