diff --git a/requirements.txt b/requirements.txt
index cd7d9f033..54d94c168 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,6 +13,7 @@ einops
 xformers
 optimum
 hf_transfer
+numba
 numpy==1.24.4
 # qlora things
 bert-score==0.3.13