Cool_Tools/ultimatevocalremovergui
1
0
Fork 0
mirror of https://github.com/Anjok07/ultimatevocalremovergui.git synced 2025-02-17 19:19:15 +01:00
Code Issues 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show Stats 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()
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')