Commit bc2e9461 authored by Phil Wang's avatar Phil Wang
Browse files

oops

parent 38175d4b

audiolm_pytorch/data.py

+4 −1
Original line number Diff line number Diff line
@@ -6,6 +6,9 @@ import torch 
from torch.nn.utils.rnn import pad_sequence

from torch.utils.data import Dataset, DataLoader



def exists(val):

    return val is not None



# dataset functions



class SoundDataset(Dataset):
@@ -32,7 +35,7 @@ class SoundDataset(Dataset): 
        file = self.files[idx]

        data, _ = sf.read(file)



        if self.seq_len_multiple_of:

        if exists(self.seq_len_multiple_of):

            mult = self.seq_len_multiple_of

            data_len = len(data)

            data = data[:(data_len // mult * mult)]

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.0.15',

  version = '0.0.16',

  license='MIT',

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

  author = 'Phil Wang',