Cool_Tools/ultimatevocalremovergui
1
0
Fork 0
mirror of https://github.com/Anjok07/ultimatevocalremovergui.git synced 2024-09-23 19:18:38 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update separate.py

Browse Source
This commit is contained in:
Anjok07 2023-10-17 21:18:41 -05:00 committed by GitHub
parent e054c545d9
commit 7036cc2ab0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
separate.py
View File

@ -762,8 +762,9 @@ class SeperateMDXC(SeperateAttributes):
chunks = mix.unfold(1, chunk_size, hop_size).transpose(0, 1)
batches = [chunks[i : i + batch_size] for i in range(0, len(chunks), batch_size)]
X = torch.zeros(S, *mix.shape).to(self.device) if S > 1 else torch.zeros_like(mix).to(self.device)
X = torch.zeros(S, *mix.shape) if S > 1 else torch.zeros_like(mix)
X = X.to(self.device)
with torch.no_grad():
cnt = 0
for batch in batches: