heimoshuiyu/ultimatevocalremovergui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Anjok07
2023-01-01 22:48:35 -06:00
committed by GitHub
parent 02186f5fb0
commit bdb55bf727
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
separate.py
View File

@@ -373,10 +373,7 @@ class SeperateDemucs(SeperateAttributes):
self.start_inference() self.start_inference()
if self.is_gpu_conversion >= 0: if self.is_gpu_conversion >= 0:
if OPERATING_SYSTEM == 'Darwin': self.device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
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')
else: else:
self.device = torch.device('cpu') self.device = torch.device('cpu')