Hugging Face's logo

Spaces:
bala1802
/
StableDiffusionModel
Sleeping

App Files Community
bala1802 commited on
Commit
5b3113b
·
verified ·
1 Parent(s): 1975737

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +2 -2
config.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
 
3
- DEVICE = "mps"
4
- # DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
5
  HEIGHT = 512
6
  WIDTH = 512
7
  GUIDANCE_SCALE = 8
 
1
  import torch
2
 
3
+ # DEVICE = "mps"
4
+ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
5
  HEIGHT = 512
6
  WIDTH = 512
7
  GUIDANCE_SCALE = 8