diff --git a/setup.py b/setup.py index d24c372..88499d6 100644 --- a/setup.py +++ b/setup.py @@ -19,10 +19,11 @@ 'contrastive learning' ], install_requires=[ + 'torch>=1.6', + 'fairseq', 'audiolm-pytorch', 'einops>=0.4', 'vector-quantize-pytorch>=0.10.15', - 'torch>=1.6', ], classifiers=[ 'Development Status :: 4 - Beta',