diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index 028ca75..a5d7614 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -27,4 +27,6 @@ termcolor
 numba
 segmentation-models-pytorch
 joblib
+timm
+deepspeed
 GDAL>=3.3.0