From aa86002f549ded6cacfa8a9b1793909db2d020c8 Mon Sep 17 00:00:00 2001 From: lmeribal Date: Fri, 18 Oct 2024 19:10:57 +0000 Subject: [PATCH] upd pyproject --- pyproject.toml | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 95596fa..2ee4b1b 100755 --- a/pyproject.toml +++ b/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"]