pyproject.toml

[tool.poetry]
name = "llm_alignment_playground"
version = "0.1.0"
description = "Fine-Tuning Engine"
authors = ["Sergei Bratchikov <hivaze.me@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "3.10.*"
jupyter = "^1.0.0"
pandas = "^2.2.1"
datasets = "^3.5.0"
tokenizers = "==0.21.0"
loguru = "^0.7.2"
pydantic = "^2.6.4"
pydantic-settings = "^2.2.1"
wandb = "^0.19.6"
evaluate = "^0.4.2"
transformers = "^4.51.0"
accelerate = "^1.6.0"
peft = "^0.12.0"
s3cmd = "^2.4.0"
clearml = "^1.16.1"
torch = "==2.5.1"
deepspeed = {platform = 'linux', version = "==0.14.5"}
sentence-transformers = "^3.0.1"
trl = "0.16.0"
vllm = {platform = 'linux', version="==0.6.5"}
xformers = {platform = 'linux', version="==0.0.28.post3"}
tabulate = "^0.9.0"
orjson = "^3.10.15"
gradio = "==4.43.0"
dvc = "==3.53.1"
liger-kernel = "^0.5.6"
huggingface-hub = "^0.30.1"
openai = "^1.44.1"
bitsandbytes = "^0.45.2"
galore-torch = "^1.0"
apollo-torch = "^1.0.3"
torchao = "^0.10.0"
ruff = "^0.9.9"
faiss-cpu = "^1.10.0"


[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"