diff --git a/requirements.txt b/requirements.txt
index 19c6c4c..69c3b30 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,6 @@ accelerate>=0.30.0
 einops
 packaging
 sentencepiece
-timm>=0.6.12
\ No newline at end of file
+timm>=0.6.12
+numpy<=1.26.4
+protobuf>=4.25.4
\ No newline at end of file