| from setuptools import setup, find_packages | |
| setup( | |
| name='neucodec', | |
| version='0.0.1', | |
| description='A package for neucodec, based on xcodec2.', | |
| long_description_content_type='text/markdown', | |
| author='Harry Julian', | |
| author_email='[email protected]', | |
| packages=find_packages(), | |
| install_requires=[ | |
| 'librosa', | |
| 'soundfile', | |
| 'numpy>=2.0.2', | |
| 'omegaconf>=2.3.0', | |
| 'torch>=2.5.1', | |
| 'torchaudio>=2.5.1', | |
| 'torchao>=0.5.0', | |
| 'torchtune>=0.3.1', | |
| 'vector-quantize-pytorch>=1.17.8', | |
| 'rotary-embedding-torch>=0.8.4', | |
| 'transformers>=4.44.2', | |
| 'boto3>1.0', | |
| 'tqdm', | |
| ], | |
| classifiers=[ | |
| 'Programming Language :: Python', | |
| 'Programming Language :: Python :: 3', | |
| 'Programming Language :: Python :: 3.10', | |
| ], | |
| ) | |