recoilme commited on
Commit
96d15bc
1 Parent(s): 8f1e053

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ dtype = torch.float32
10
  if torch.cuda.is_available():
11
  device = "cuda"
12
  dtype = torch.float16
 
13
 
14
  # check if MPS is available OSX only M1/M2/M3 chips
15
  mps_available = hasattr(torch.backends, "mps") and torch.backends.mps.is_available()
 
10
  if torch.cuda.is_available():
11
  device = "cuda"
12
  dtype = torch.float16
13
+
14
 
15
  # check if MPS is available OSX only M1/M2/M3 chips
16
  mps_available = hasattr(torch.backends, "mps") and torch.backends.mps.is_available()