upd pyproject

pyproject.toml

@@ -13,12 +13,12 @@ keywords = ["alignment", "llm", "dpo", "ppo", "rlhf"]
 
 [tool.poetry.dependencies]
 python = "^3.10"
-transformers = "4.43.1"
+transformers = "4.42.0"
 peft = "0.8.2"
 typer ="^0.9.0"
 wandb ="^0.15.3"
 scipy = "^1.1.0"
-torch = "2.3.1"
+torch = "2.1.2"
 numpy = "^1.21"
 tomlkit ="^0.11.8"
 platformdirs ="^3.5.3"
@@ -42,12 +42,14 @@ pydantic ="^2.7.0"
 langchain-huggingface = "^0.0.3"
 clearml = "^1.16.4"
 cached-path = "^1.6.3"
-timm = "^0.9.7"
-pytorchvideo = "^0.1.5"
-torchaudio = "^2.0.2"
-soundfile = "^0.12.1"
-albumentations = "^1.3.1"
+timm ="^0.9.7"
+pytorchvideo ="^0.1.5"
+torchaudio ="^2.0.2"
+soundfile ="^0.12.1"
+albumentations ="^1.3.1"
 opencv-python = "^4.10.0.84"
+xformers = { version = "^0.0.23.post1", optional = true }
+accelerate = { version = "0.27", optional = true }
 
 faiss-gpu = { version = "^1.7.2", optional = true }
 deepspeed = { version = "0.12", optional = true }
@@ -70,7 +72,7 @@ annotated-types = "^0.6.0"
 
 [tool.poetry.extras]
 deepspeed = ["deepspeed", "accelerate"]
-gpu = ["faiss-gpu", "xformers", "vllm"]
+gpu = ["faiss-gpu", "vllm"]
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]