[project]
name = "ai-lab"
version = "0.1.0"
description = "Lab for testing different uses of transformers"
authors = [{ name = "Cyril", email = "decostanzicyril@gmail.com" }]

[tool.poetry]
name = "ai-lab"
version = "0.1.0"
description = "Lab for testing different uses of transformers"
authors = ["Cyril"]
packages = [{ include = "src" }]

[tool.poetry.dependencies]
python = ">=3.12,<3.14"
torch = "^2.0.0"
transformers = "^4.30.0"
tokenizers = "^0.13.0"
numpy = "^1.24.0"
accelerate = "^0.20.0"
fastapi = "^0.104.0"
uvicorn = { extras = ["standard"], version = "^0.24.0" }
pydantic = "^2.5.0"

[tool.poetry.scripts]
ai-lab = "src.main:main"

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