[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"