diff --git a/setup.py b/setup.py
index 993c723..8bf22e6 100755
--- a/setup.py
+++ b/setup.py
@@ -58,6 +58,7 @@ setup(
         "vocos",
         "h5py",
         "torchscale @ git+https://git.ecker.tech/mrq/torchscale",
+        "prodigyopt @ git+https://github.com/konstmish/prodigy",
     ],
     url="https://git.ecker.tech/mrq/vall-e",
 )