NikkeDoy/SISU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚗️ | Small fixes here and there

Browse Source
This commit is contained in:
NikkeDoy
2025-12-11 23:06:38 +02:00
parent e3e555794e
commit 571c403b93
4 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
AudioUtils.py
View File

@@ -8,13 +8,12 @@ def stereo_tensor_to_mono(waveform: torch.Tensor) -> torch.Tensor:
def pad_tensor(audio_tensor: torch.Tensor, target_length: int = 512) -> torch.Tensor:
padding_amount = target_length - audio_tensor.size(-1)
current = audio_tensor.size(-1)
padding_amount = target_length - current
if padding_amount <= 0:
return audio_tensor
padded_audio_tensor = F.pad(audio_tensor, (0, padding_amount))
return padded_audio_tensor
return F.pad(audio_tensor, (0, padding_amount))
def split_audio(audio_tensor: torch.Tensor, chunk_size: int = 512, pad_last_tensor: bool = False) -> list[torch.Tensor]: