Update setup.

LCAV · Apr 29, 2024 · 8ee5ce1 · 8ee5ce1
1 parent 538a721
commit 8ee5ce1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/recon_requirements.txt b/recon_requirements.txt
@@ -10,4 +10,5 @@ torch >= 2.0.0
 torchvision
 torchmetrics
 lpips
-wandb
+wandb
+datasets
diff --git 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"]},
 )