fix windows issue with uname

MannLabs · May 31, 2024 · a77e5e8 · a77e5e8
1 parent 11628ab
commit a77e5e8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/alphadia/libtransform.py b/alphadia/libtransform.py
@@ -301,7 +301,11 @@ def forward(self, input: SpecLibBase) -> SpecLibBase:
         # Check if CPU or GPU/MPS should be used
         device = "cpu"
         if self.use_gpu:
-            device = "mps" if os.uname().sysname == "Darwin" else "gpu"
+            try:
+                device = "mps" if os.uname().sysname == "Darwin" else "gpu"
+            except AttributeError:
+                # Windows does not support uname
+                device = "gpu"
 
         model_mgr = ModelManager(device=device)
         if self.checkpoint_folder_path is not None: