mirror of
https://github.com/wassname/SimPO.git
synced 2026-06-27 19:13:33 +08:00
wassname
31 lines
553 B
TOML
31 lines
553 B
TOML
[project]
|
||
name = "simpo"
|
||
version = "0.1.0"
|
||
description = ""
|
||
readme = "README.md"
|
||
requires-python = ">=3.10.9,<4.0"
|
||
|
||
dependencies = [
|
||
"torch>=2.1.2",
|
||
"deepspeed>=0.14",
|
||
"wandb>=0.18.2",
|
||
# git‐based dep
|
||
"alignment-handbook",
|
||
]
|
||
|
||
[dependency-groups]
|
||
dev = [
|
||
"pytest",
|
||
"ipykernel>=6.29.5",
|
||
]
|
||
|
||
[build-system]
|
||
requires = ["hatchling"]
|
||
build-backend = "hatchling.build"
|
||
|
||
[tool.hatch.build.targets.wheel]
|
||
packages = ["simpo"]
|
||
|
||
[tool.uv.sources]
|
||
alignment-handbook = { git = "https://github.com/huggingface/alignment-handbook.git" }
|