Spaces:

anvilarth
/

Garage

Runtime error

anvilarth commited on Sep 24, 2024

Commit

fd7ecc6

verified ·

1 Parent(s): 8d2f655

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,8 +47,7 @@ class GradioWindow():
         self.SAM_CHECKPOINT_PATH = SAM_CHECKPOINT_PATH
         # self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        self.device = "cuda"
-        # self.device = "cpu"
         # for debug
         # self.augmenter = None
         self.augmenter = Augmenter(device=self.device)

         self.SAM_CHECKPOINT_PATH = SAM_CHECKPOINT_PATH
         # self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.device = "cpu"
         # for debug
         # self.augmenter = None
         self.augmenter = Augmenter(device=self.device)