cocktailpeanut commited on
Commit
98c1da8
1 Parent(s): af428bc
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ if torch.cuda.is_available():
25
  dtype = torch.float16
26
  elif torch.backends.mps.is_available():
27
  device = "mps"
28
- #dtype = torch.float32
29
  else:
30
  device = "cpu"
31
  dtype = torch.float32
 
25
  dtype = torch.float16
26
  elif torch.backends.mps.is_available():
27
  device = "mps"
28
+ dtype = torch.float16
29
  else:
30
  device = "cpu"
31
  dtype = torch.float32