pyproject.toml

[project]
name = "wolfsbit-01-training"
version = "0.1.0"
description = "LoRA fine-tuning for lupus activity scoring"
requires-python = ">=3.12"
dependencies = [
    "accelerate>=1.12.0",
    "coremltools>=9.0",
    "datasets>=4.4.1",
    "huggingface-hub[cli]>=0.36.0",
    "mistral-common>=1.8.6",
    "numpy>=2.3.5",
    "peft",
    "torch>=2.9.1",
    "torchaudio>=2.9.1",
    "torchvision>=0.24.1",
    "transformers>=4.57.3",
]

[tool.uv]
package = false

[tool.uv.sources]
peft = { git = "https://github.com/huggingface/peft.git" }
transformers = { git = "https://github.com/huggingface/transformers.git" }