From bdb55bf7279e8c81e3af49f014334490c47e4196 Mon Sep 17 00:00:00 2001 From: Anjok07 <68268275+Anjok07@users.noreply.github.com> Date: Sun, 1 Jan 2023 22:48:35 -0600 Subject: [PATCH] Add files via upload --- separate.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/separate.py b/separate.py index e0231ad..45c94b2 100644 --- a/separate.py +++ b/separate.py @@ -373,10 +373,7 @@ class SeperateDemucs(SeperateAttributes): self.start_inference() if self.is_gpu_conversion >= 0: - if OPERATING_SYSTEM == 'Darwin': - self.device = torch.device('mps' if torch.backends.mps.is_available() and self.demucs_version in [DEMUCS_V1, DEMUCS_V2] and 'Tasnet' not in self.model_name else 'cpu') - else: - self.device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu') + self.device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu') else: self.device = torch.device('cpu')