diff --git a/setup.py b/setup.py
index 3402d9614..05fc8406e 100644
--- a/setup.py
+++ b/setup.py
@@ -30,6 +30,9 @@
         'setuptools',
         'scikit-learn',
         'timm',
+        'omegaconf',
+        'lightning',
+        'setuptools==68.2.2'
     ],
     classifiers=[
         'Development Status :: 3 - Alpha',