⬆️ Style Transfer: Upgrade and pin some dependencies

arkhn · Feb 28, 2024 · 6cd29c4 · 6cd29c4
1 parent e0b48f7
commit 6cd29c4
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 15 deletions.
diff --git a/lib/style-transfer/poetry.lock b/lib/style-transfer/poetry.lock
diff --git a/lib/style-transfer/pyproject.toml b/lib/style-transfer/pyproject.toml
@@ -59,17 +59,18 @@ deepspeed-mii = "^0.2.1"
 fschat = {extras = ["model-worker", "webui"], version = "^0.2.35"}
 hydra-colorlog = "^1.2.0"
 hydra-core = "^1.3.2"
+hydra-joblib-launcher = "^1.2.0"
 peft = "^0.7.1"
 pytest = "^7.4.2"
 pytest-cov = "^4.1.0"
 python = "~3.9"
 quickumls = "^1.4.1"
 rich = "^13.5.2" # beautiful text formatting in terminal
-sentence-transformers = "^2.2.2"
+sentence-transformers = "^2.3.1"
 spacy = "^3.6.1"
 torch = "^2.0.1"
 torchmetrics = "^1.1.2"
-transformers = "^4.37.2"
+transformers = "4.36.2"
 trl = "^0.7.9"
 wandb = "^0.16.3"