diff --git a/requirements-onnx.txt b/requirements-onnx.txt
index 976591f7..6bf232d7 100644
--- a/requirements-onnx.txt
+++ b/requirements-onnx.txt
@@ -3,6 +3,7 @@
 # See instructions at https://pytorch.org/get-started/previous-versions/
 
 click
+einops
 h5py
 librosa<0.10.0
 lightning~=2.1.0
diff --git a/requirements.txt b/requirements.txt
index 90f3f9c5..5701b85b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,6 +3,7 @@
 # See instructions at https://pytorch.org/get-started/locally/
 
 click
+einops
 h5py
 librosa<0.10.0
 lightning~=2.3.0