[tool.poetry]
name = "bad-gpt"
version = "0.1.0"
description = ""
authors = []
readme = "README.md"
package-mode = false

[tool.poetry.dependencies]
python = "^3.10"
torch = { version = "^2.3.0", source = "pytorch-gpu" }
numpy = "^1.26.4"
datasets = "^2.19.0"
tiktoken = "^0.6.0"


[tool.poetry.group.dev.dependencies]
ipykernel = "^6.29.4"


[[tool.poetry.source]]
name = "pytorch-gpu"
url = "https://download.pytorch.org/whl/cu118"
priority = "supplemental"

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