Skip to content

Commit

Permalink
update version
Browse files Browse the repository at this point in the history
  • Loading branch information
Малахов Алексей Павлович committed Sep 19, 2024
1 parent dfd8ff3 commit 99d4861
Showing 1 changed file with 6 additions and 8 deletions.
14 changes: 6 additions & 8 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -39,20 +39,19 @@ pydantic ="^2.7.0"
langchain-huggingface = "^0.0.3"
clearml = "^1.16.4"
cached-path = "^1.6.3"
opencv-python = "^4.10.0.84"
albumentations = "^1.3.1"
soundfile = "^0.12.1"
torchaudio = "^2.0.2"
pytorchvideo = "^0.1.5"
timm = "^0.9.7"

faiss-gpu = { version = "^1.7.2", optional = true }
xformers = {version = "0.0.27.post2", optional = true}
vllm = { version = "0.5.4", optional = true }
deepspeed = {version = "^0.15.0", optional = true}
accelerate = {version = "0.33", optional = true}

opencv-python = { version = "^4.10.0.84", optional = true }
albumentations = { version = "^1.3.1", optional = true }
soundfile = { version = "^0.12.1", optional = true }
torchaudio = { version = "^2.0.2", optional = true }
pytorchvideo = { version = "^0.1.5", optional = true }
timm = { version = "^0.9.7", optional = true }

[tool.poetry.dev-dependencies]
coverage = "^7.2.7"
pytest = "^7.3.2"
Expand All @@ -69,7 +68,6 @@ annotated-types = "^0.6.0"
[tool.poetry.extras]
deepspeed = ["deepspeed", "accelerate"]
gpu = ["faiss-gpu", "xformers", "vllm"]
multimodal = ["opencv-python", "albumentations", "soundfile", "pytorchvideo", "timm"]

[build-system]
requires = ["poetry-core>=1.0.0"]
Expand Down

0 comments on commit 99d4861

Please sign in to comment.