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
7036cc2ab0
commit
b20ab25080
@ -757,7 +757,6 @@ class SeperateMDXC(SeperateAttributes):
|
||||
mix_shape = mix.shape[1]
|
||||
pad_size = hop_size - (mix_shape - chunk_size) % hop_size
|
||||
mix = torch.cat([torch.zeros(2, chunk_size - hop_size), mix, torch.zeros(2, pad_size + chunk_size - hop_size)], 1)
|
||||
mix = mix.to(self.device)
|
||||
|
||||
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)]
|
||||
@ -769,7 +768,7 @@ class SeperateMDXC(SeperateAttributes):
|
||||
cnt = 0
|
||||
for batch in batches:
|
||||
self.running_inference_progress_bar(len(batches))
|
||||
x = model(batch)
|
||||
x = model(batch.to(self.device))
|
||||
|
||||
for w in x:
|
||||
X[..., cnt * hop_size : cnt * hop_size + chunk_size] += w
|
||||
|
Loading…
Reference in New Issue
Block a user