mirror of
https://github.com/Anjok07/ultimatevocalremovergui.git
synced 2024-11-12 01:50:48 +01:00
Update separate.py
This commit is contained in:
parent
e054c545d9
commit
7036cc2ab0
@ -762,7 +762,8 @@ 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
|
||||
|
Loading…
Reference in New Issue
Block a user