diff --git a/scripts/requirements.txt b/scripts/requirements.txt
index 675faf6..68b50f0 100644
--- a/scripts/requirements.txt
+++ b/scripts/requirements.txt
@@ -8,7 +8,7 @@ tokenizers>=0.13.0
 numpy>=1.24.0
 requests>=2.25.0
 tqdm>=4.60.0
-model2vec>=0.1.0
+model2vec[distill]>=0.1.0
 sentence-transformers>=2.2.0
 sentencepiece>=0.1.97
 einops>=0.8.1