@ -60,6 +60,7 @@ sentence-transformers = "*"
peft = "*"
psutil = "*"
timm = "*"
supervision = "*"
[tool.poetry.dev-dependencies]
black = "23.3.0"
@ -33,3 +33,4 @@ pre-commit==3.6.2
psutil
black
tenacity
supervision