Update dependencies in pyproject.toml

2024-11-13 13:56:00 +05:30
parent 19fa538748
commit bf33d8fd97
1 changed files with 12 additions and 8 deletions
@@ -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"
 ]