JasonYinnnn commited on
Commit
c022cd6
·
1 Parent(s): 963d16c

run_segmentation on CPU

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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