diff --git a/pyproject.toml b/pyproject.toml
index 888dbfd..de25c46 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,6 +20,8 @@ dependencies = [
     "torch",
     "accelerate==1.6.0",
     "transformers==4.51.3", # we develop this project on transformers==4.51.3, later version may not be compatible
+    "llvmlite>=0.40.0",
+    "numba>=0.57.0",
     "diffusers",
     "tqdm",
     "numpy",