⬆️ Style Transfer: bump vllm and bnb versions

arkhn · Oct 10, 2024 · 9a77b56 · 9a77b56
1 parent 9af48b2
commit 9a77b56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/style-transfer/pyproject.toml b/lib/style-transfer/pyproject.toml
@@ -51,7 +51,7 @@ repository = "https://github.com/arkhn/open-nlp"
 version = "0.1.0"
 
 [tool.poetry.dependencies]
-bitsandbytes = "^0.43.1"
+bitsandbytes = "0.44.1"
 datasets = "^2.16.1"
 hydra-colorlog = "^1.2.0"
 hydra-core = "^1.3.2"
@@ -66,7 +66,7 @@ torch = "^2.3.0"
 tqdm = "^4.66.1"
 transformers = "4.45.1"
 trl = "^0.9.6"
-vllm = "^0.5.0.post1"
+vllm = "0.6.2"
 wandb = "^0.16.3"
 
 [tool.poetry.dev-dependencies]