[project]
dependencies = [
  "absl-py>=2.1.0",
  "accelerate==1.2.1",
  "datasets==3.2.0",
  "deepspeed==0.16.2",
  "evaluate==0.4.3",
  "huggingface-hub==0.30.1",
  "librosa>=0.10.2.post1",
  "markupsafe==2.1.5",
  "ms-swift>=1.3.0",
  "nltk>=3.9.1",
  "numba>=0.60.0",
  "peft==0.14.0",
  "pip==24.3.1",
  "pre-commit>=4.0.1",
  "pytest>=8.3.4",
  "requests==2.32.3",
  "rouge-score>=0.1.2",
  "safetensors>=0.5.2",
  "setuptools>=70.0.0",
  "soundfile>=0.13.0",
  "torch==2.5.1+cu124",
  "torchaudio==2.5.1+cu124",
  "torchvision==0.20.1+cu124",
  "transformers==4.48.0",
  "trl==0.13.0",
  "wandb>=0.19.4",
  "wheel>=0.45.1",
]
description = "Add your description here"
name = "cl-lmm"
readme = "README.md"
requires-python = ">=3.11"
version = "0.1.0"

[project.optional-dependencies]
compile = ["flash-attn>=2.7.2.post1"]

[tool.uv.sources]
markupsafe = { index = "pytorch" }
requests = { index = "pypi" }
torch = { index = "pytorch" }
torchaudio = { index = "pytorch" }
torchvision = { index = "pytorch" }

[[tool.uv.index]]
name = "pypi"
url = "https://pypi.org/simple"

[tool.uv]
no-build-isolation-package = ["flash-attn"]
concurrent-builds = 4

[[tool.uv.index]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu124"

[tool.uv.workspace]
members = ["ms-swift"]

[tool.black]
line-length = 88
exclude = "transformers_repo|peft_repo|.venv"
target-version = ["py311"]

[tool.pytest.ini_options]
addopts = ["--color=yes", "--durations=0", "-v", "--capture=tee-sys"]
norecursedirs = ["src/transformers_repo", "src/peft_repo", ".venv"]