Spaces:

bala1802
/

StableDiffusionModel

Sleeping

bala1802 commited on Jan 13

Commit

5b3113b

•

1 Parent(s): 1975737

Update config.py

Files changed (1) hide show

config.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import torch
-DEVICE = "mps"
-# DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 HEIGHT = 512
 WIDTH = 512
 GUIDANCE_SCALE = 8

 import torch
+# DEVICE = "mps"
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 HEIGHT = 512
 WIDTH = 512
 GUIDANCE_SCALE = 8