Commit 7f788a62 authored by Phil Wang's avatar Phil Wang
Browse files

also allow for single resample target hz for sounddataset

parent 5d6cd874

audiolm_pytorch/data.py

+1 −1
Original line number Diff line number Diff line
@@ -33,7 +33,7 @@ class SoundDataset(Dataset): 
        folder,

        exts = ['flac', 'wav'],

        max_length = None,

        target_sample_hz: Optional[Tuple[Optional[int], ...]] = None,

        target_sample_hz: Optional[Union[int, Tuple[Optional[int], ...]]] = None,

        seq_len_multiple_of = None

    ):

        super().__init__()

setup.py

+1 −1
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages 
setup(

  name = 'audiolm-pytorch',

  packages = find_packages(exclude=[]),

  version = '0.4.3',

  version = '0.4.4',

  license='MIT',

  description = 'AudioLM - Language Modeling Approach to Audio Generation from Google Research - Pytorch',

  author = 'Phil Wang',