mumble-cleanup / pyproject.toml
adikuma's picture
initial upload: cleanup code and 688-pair seed dataset
fd0b01f verified
Raw
History Blame Contribute Delete
818 Bytes
[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",
]