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

will need ema for soundstream

parent 9d832142

setup.py

+1 −0
Original line number Diff line number Diff line
@@ -19,6 +19,7 @@ setup( 
  ],

  install_requires=[

    'einops>=0.4',

    'ema-pytorch',

    'torch>=1.6',

    'vector-quantize-pytorch'

  ],