[project]
name = "cleanup"
version = "0.1.0"
description = "Fine-tuned Qwen2.5-0.5B-Instruct LoRA adapter that cleans Mumble dictation transcripts."
readme = "README.md"
requires-python = ">=3.11,<3.12"
dependencies = [
    "torch>=2.4",
    "transformers>=4.46,<5",
    "peft>=0.13",
    "trl>=0.12",
    "datasets>=3.0",
    "accelerate>=1.0",
    "jiwer>=3.0",
    "optimum[onnxruntime]>=1.23",
    "onnxruntime>=1.19",
    "onnx>=1.16",
    "matplotlib>=3.9",
    "pyyaml>=6.0",
    "tqdm>=4.66",
    "huggingface_hub>=0.25",
    "numpy>=1.26,<2.2",
    "sentencepiece>=0.2",
    "protobuf>=4.25,<5",
    "Levenshtein>=0.25",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["src/cleanup"]

[dependency-groups]
dev = [
    "pytest>=8.0",
]