compatibility with older torch

Amshaker · Jul 18, 2023 · 3371c72 · 3371c72
1 parent 10371a7
commit 3371c72
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 7 deletions.
diff --git a/inference/interact/fbrs/controller.py b/inference/interact/fbrs/controller.py
@@ -1,5 +1,8 @@
 import torch
-from torch import mps
+try:
+    from torch import mps
+except:
+    pass
 
 from ..fbrs.inference import clicker
 from ..fbrs.inference.predictors import get_predictor

diff --git a/inference/interact/gui.py b/inference/interact/gui.py
@@ -21,7 +21,10 @@
 
 import numpy as np
 import torch
-from torch import mps
+try:
+    from torch import mps
+except:
+    print('torch.MPS not available.')
 
 from PyQt6.QtWidgets import (QWidget, QApplication, QComboBox, QCheckBox,
     QHBoxLayout, QLabel, QPushButton, QTextEdit, QSpinBox, QFileDialog,

diff --git a/inference/interact/interactive_utils.py b/inference/interact/interactive_utils.py
@@ -26,11 +26,14 @@ def index_numpy_to_one_hot_torch(mask, num_classes):
 """
 Some constants fro visualization
 """
-if torch.cuda.is_available():
-    device = torch.device("cuda")
-elif torch.backends.mps.is_available():
-    device = torch.device("mps")
-else:
+try:
+    if torch.cuda.is_available():
+        device = torch.device("cuda")
+    elif torch.backends.mps.is_available():
+        device = torch.device("mps")
+    else:
+        device = torch.device("cpu")
+except:
     device = torch.device("cpu")
 
 color_map_np = np.frombuffer(davis_palette, dtype=np.uint8).reshape(-1, 3).copy()