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

need latest version of torch for complex stft discriminator distributed training

parent c8fe964a

setup.py

+2 −2
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.11.16',

  version = '0.11.17',

  license='MIT',

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

  author = 'Phil Wang',
@@ -28,7 +28,7 @@ setup( 
    'Mega-pytorch',

    'scikit-learn',

    'sentencepiece',

    'torch>=1.6',

    'torch>=1.12',

    'torchaudio',

    'transformers',

    'tqdm',