71 lines
1.5 KiB
TOML
71 lines
1.5 KiB
TOML
[project]
|
|
dependencies = [
|
|
"absl-py>=2.1.0",
|
|
"accelerate==1.2.1",
|
|
"calflops>=0.3.2",
|
|
"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.6.0",
|
|
"torchaudio==2.6.0",
|
|
"torchvision==0.21.0",
|
|
"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"]
|