diff --git a/requirements.txt b/requirements.txt
index 1289318..926ef41 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,9 +3,14 @@ diffusers[torch]==0.30.2
 diffusionkit==0.4.0
 torch
 transformers==4.44.2
+tokenizers==0.15.2
+sentencepiece==0.1.99
 scipy
 scikit-learn
 pytest
 invisible-watermark
 safetensors
 matplotlib
+wheel
+setuptools>=65.5.1
+ninja