mirror of
https://github.com/wassname/alignment-handbook.git
synced 2026-06-27 14:43:36 +08:00
wassname
63 lines
1.3 KiB
TOML
63 lines
1.3 KiB
TOML
[project]
|
|
name = "alignment-handbook"
|
|
version = "0.1.0"
|
|
description = "Add your description here"
|
|
readme = "README.md"
|
|
requires-python = ">=3.10"
|
|
|
|
dependencies = [
|
|
"accelerate>=0.29.2",
|
|
"bitsandbytes>=0.43.0",
|
|
"black>=24.4.2",
|
|
"datasets>=2.18.0",
|
|
"deepspeed>=0.14.4",
|
|
"einops>=0.6.1",
|
|
"evaluate==0.4.0",
|
|
"flake8>=6.0.0",
|
|
"hf-doc-builder>=0.4.0",
|
|
"hf_transfer>=0.1.4",
|
|
"huggingface-hub>=0.19.2,<1.0",
|
|
"isort>=5.12.0",
|
|
"ninja>=1.11.1",
|
|
"numpy>=1.24.2",
|
|
"packaging>=23.0",
|
|
"parameterized>=0.9.0",
|
|
"peft>=0.9.0",
|
|
"protobuf<=3.20.2", # Needed to avoid conflicts with `transformers`
|
|
"pytest",
|
|
"safetensors>=0.3.3",
|
|
"sentencepiece>=0.1.99",
|
|
"scipy",
|
|
"tensorboard",
|
|
"torch>=2.1.2",
|
|
"transformers>=4.39.3",
|
|
"trl>=0.9.6,<0.13.0",
|
|
"jinja2>=3.0.0",
|
|
"tqdm>=4.64.1",
|
|
"wheel>=0.45.1",
|
|
"setuptools>=80.9.0",
|
|
"hatchling>=1.27.0",
|
|
"editables>=0.5",
|
|
"flash-attn>=2.7.4.post1",
|
|
"wandb>=0.19.11",
|
|
"loguru>=0.7.3",
|
|
]
|
|
|
|
|
|
[dependency-groups]
|
|
dev = [
|
|
"pytest",
|
|
"ipykernel>=6.29.5",
|
|
"ipywidgets>=8.1.7",
|
|
]
|
|
|
|
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
packages = ["src/alignment"]
|
|
|
|
[tool.uv.pip]
|
|
no-build-isolation-package = ["flash-attn"]
|