diff --git a/recon_requirements.txt b/recon_requirements.txt
index 309d71fe..78dd418d 100644
--- a/recon_requirements.txt
+++ b/recon_requirements.txt
@@ -10,4 +10,5 @@ torch >= 2.0.0
 torchvision
 torchmetrics
 lpips
-wandb
\ No newline at end of file
+wandb
+datasets
\ No newline at end of file
diff --git a/setup.py b/setup.py
index d1ab6d68..c3d0fc2c 100644
--- a/setup.py
+++ b/setup.py
@@ -34,6 +34,7 @@
         "rawpy>=0.16.0",  # less than python 3.12
         "paramiko>=3.2.0",
         "hydra-core",
+        "slm_controller @ git+https://github.com/ebezzam/slm-controller.git"
     ],
     extra_requires={"dev": ["pudb", "black"]},
 )