lsb commited on
Commit
182dec8
1 Parent(s): 829ed2e

fp32 for now

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from datetime import datetime
13
  # preferred_device = "cuda" if torch.cuda.is_available() else ("mps" if torch.backends.mps.is_available() else "cpu")
14
  # but segformer does not work on mps lolololol
15
  preferred_device = "cuda" if torch.cuda.is_available() else "cpu"
16
- preferred_dtype = torch.float16 if preferred_device == 'cuda' else torch.float32
17
 
18
  seg_model_img_size = 768
19
  seg_model_size = 0
 
13
  # preferred_device = "cuda" if torch.cuda.is_available() else ("mps" if torch.backends.mps.is_available() else "cpu")
14
  # but segformer does not work on mps lolololol
15
  preferred_device = "cuda" if torch.cuda.is_available() else "cpu"
16
+ preferred_dtype = torch.float32 # torch.float16 if preferred_device == 'cuda' else torch.float32
17
 
18
  seg_model_img_size = 768
19
  seg_model_size = 0