pyproject.toml

[project]
name = "model-experiments"
version = "0.1.0"
description = "Using models as classifiers and changing their innards."
readme = "README.md"
requires-python = ">=3.14"
dependencies = [
    "accelerate>=1.13.0",
    "bitsandbytes>=0.43.0",
    "hf>=1.18.0",
    "labl-utils",
    "matplotlib>=3.9",
    "python-dotenv>=1.1.0",
    "scikit-learn>=1.5",
    "torch>=2.11.0",
    "torchaudio>=2.11.0",
    "torchvision>=0.26.0",
    "transformers>=5.10.2",
]

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true

[tool.uv.sources]
torch = { index = "pytorch-cu128" }
torchvision = { index = "pytorch-cu128" }
torchaudio = { index = "pytorch-cu128" }
labl-utils = { git = "https://github.com/bentdixon/labl-utils" }