Commit 8ebbb2e9 authored by Phil Wang's avatar Phil Wang
Browse files

fix bug with collation

parent 60e4cd84

audiolm_pytorch/data.py

+2 −1
Original line number Diff line number Diff line
@@ -92,7 +92,8 @@ def collate_one_or_multiple_tensors(fn): 
        is_one_data = not isinstance(data[0], tuple)



        if is_one_data:

            data = (data,)

            data = torch.stack(data)

            return (data,)



        outputs = []

        for datum in zip(*data):

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

  version = '0.1.19',

  license='MIT',

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

  author = 'Phil Wang',