diff --git a/pyproject.toml b/pyproject.toml
index 757bf20..2b8d1c0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,18 +7,22 @@ requires-python = ">=3.10"
 dependencies = [
     "torch>=2.0.0",
     "torchvision>=0.15.0",
-    "lightning>=2.0.0",
-    "litserve>=0.1.0",
-    "wandb>=0.15.0",
-    "pydantic>=2.0.0",
-    "pydantic-settings>=2.0.0",
+    "opencv-python==4.10.0.84",
+    "easydict==1.13",
+    "einops==0.8.0",
+    "open-clip-torch==2.20.0",
+    "xformers==0.0.21",
+    "fairscale==0.4.13",
+    "torchsde==0.2.6",
+    "pytorch-lightning==2.0.1",
+    "diffusers==0.30.0",
+    "huggingface_hub==0.23.3",
+    "numpy==1.24",
     "transformers>=4.30.0",
     "accelerate>=0.20.0",
     "safetensors>=0.3.0",
-    "diffusers@ git+https://github.com/huggingface/diffusers",
     "sentencepiece",
-    "bitsandbytes",
-
+    "bitsandbytes"
 ]