fp32 for now
Browse files
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
|