47 lines
945 B
TOML
47 lines
945 B
TOML
[project]
|
|
dependencies = [
|
|
"accelerate==1.2.1",
|
|
"datasets==3.2.0",
|
|
"deepspeed==0.16.2",
|
|
"evaluate==0.4.3",
|
|
"markupsafe==2.1.5",
|
|
"peft==0.14.0",
|
|
"pip==24.3.1",
|
|
"requests==2.32.3",
|
|
"setuptools>=70.0.0",
|
|
"torch==2.5.1+cu124",
|
|
"torchaudio==2.5.1+cu124",
|
|
"torchvision==0.20.1+cu124",
|
|
"transformers==4.46.1",
|
|
"trl==0.13.0",
|
|
"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"]
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch"
|
|
url = "https://download.pytorch.org/whl/cu124"
|