[tool.poetry]
name = "llmlib"
version = "0.1.0"
description = ""
authors = ["Tomas Ruiz <tomas.ruiz.te@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.11"
bugsnag = "^4.7.1"
decord = "^0.6.0"
google-cloud-aiplatform = "^1.64"
# I cannot add the dependencies below. I suspect that torch is a build-time dependency for flash-attn, or something like that.
# transformers = "^4.44.2"
# accelerate = "^0.34.2"
# flash-attn = "^2.6.3"
# torch = "^2.4.0"

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