Spaces:
Running on Zero
Running on Zero
Commit ·
c022cd6
1
Parent(s): 963d16c
run_segmentation on CPU
Browse files
app.py
CHANGED
|
@@ -188,7 +188,8 @@ def run_segmentation(
|
|
| 188 |
|
| 189 |
global work_space
|
| 190 |
|
| 191 |
-
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 192 |
sam_segmentator.to(device=device, dtype=DTYPE if device == 'cuda' else torch.float32)
|
| 193 |
gr.Info('in run_segmentation, move sam')
|
| 194 |
|
|
|
|
| 188 |
|
| 189 |
global work_space
|
| 190 |
|
| 191 |
+
# device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 192 |
+
device = "cpu"
|
| 193 |
sam_segmentator.to(device=device, dtype=DTYPE if device == 'cuda' else torch.float32)
|
| 194 |
gr.Info('in run_segmentation, move sam')
|
| 195 |
|