Skip to content

Commit 78bbf70

Browse files
committed
Add options from old loader
1 parent 86c40b8 commit 78bbf70

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/torchaudio/utils/__init__.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,17 @@
44
from .download import download_asset
55
import os
66

7-
def load_torchcodec(file, **args):
7+
def load_torchcodec(file, normalize=True, channels_first=True, **args):
8+
if not normalize:
9+
raise Exception("Torchcodec does not support non-normalized file reading")
810
from torchcodec.decoders import AudioDecoder
911
decoder = AudioDecoder(file)
1012
if 'start_seconds' in args or 'stop_seconds' in args:
1113
samples = decoder.get_samples_played_in_range(**args)
1214
else:
1315
samples = decoder.get_all_samples()
14-
return (samples.data, samples.sample_rate)
16+
data = samples.data if channels_first else samples.data.T
17+
return (data, samples.sample_rate)
1518

1619
__all__ = [
1720
"load_torchcodec",

0 commit comments

Comments
 (0)