28 lines
631 B
TOML
28 lines
631 B
TOML
[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"
|
|
|
|
[tool.poetry.scripts]
|
|
ai-lab = "src.main:main"
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|