fine-tuning-playground / pyproject.toml
dsmueller's picture
Add new dependencies and update existing ones
feee6eb
raw
history blame contribute delete
474 Bytes
[tool.poetry]
name = "fine-tuning-playground"
version = "0.1.0"
description = ""
authors = ["dsmueller <dsmueller3760@gmail.com>"]
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.11"
trl = "^0.7.6"
datasets = "^2.16.0"
transformers = "^4.36.2"
torch = "^2.1.2"
ipykernel = "^6.27.1"
peft = "^0.7.1"
bitsandbytes = "^0.41.3.post2"
evaluate = "^0.4.1"
scikit-learn = "^1.3.2"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"