fix for ema

cmu-mlsp · Mar 23, 2023 · 2676200 · 2676200
1 parent 0491eaa
commit 2676200
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/audiolm_pytorch/version.py b/audiolm_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.24.0'
+__version__ = '0.24.1'
diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@
     'accelerate',
     'beartype',
     'einops>=0.6',
-    'ema-pytorch',
+    'ema-pytorch>=0.2.2',
     'fairseq',
     'joblib',
     'lion-pytorch',