Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

chore: update trl to grpo_vllm branch, move lighteval to uv #30

Open
wants to merge 5 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
111 changes: 111 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,111 @@
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "open-r1"
version = "0.1.0.dev0"
description = "Open R1"
readme = "README.md"
requires-python = ">=3.10.9"
license = { text = "Apache-2.0" }
authors = [
{ name = "The Hugging Face team (past and future)", email = "[email protected]" },
]
keywords = ["llm", "inference-time", "compute", "reasoning"]
classifiers = [
"Development Status :: 3 - Alpha",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.10",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
"accelerate>=1.2.1",
"datasets>=3.2.0",
"deepspeed==0.15.4",
"distilabel[vllm,ray,openai]>=1.5.2",
"einops>=0.8.0",
"hf_transfer>=0.1.4",
"huggingface-hub[cli]>=0.19.2,<1.0",
"liger_kernel==0.5.2",
"math-verify>=0.1.0",
"packaging>=23.0",
"safetensors>=0.3.3",
"sentencepiece>=0.1.99",
"torch>=2.5.1",
"vllm==0.6.6.post1",
"wandb>=0.19.1",
]

[project.optional-dependencies]
tests = ["pytest>=7.0.0", "parameterized>=0.9.0"]
quality = ["black>=24.4.2", "isort>=5.12.0", "flake8>=6.0.0"]
eval = ["lighteval>=0.1.0", "math-verify>=0.1.0"]
quantization = ["bitsandbytes==0.43.0"]
dev = [
"black>=24.4.2",
"isort>=5.12.0",
"flake8>=6.0.0",
"pytest>=7.0.0",
"parameterized>=0.9.0",
"lighteval>=0.1.0",
"math-verify>=0.1.0",
]

[tool.hatch.build]
packages = ["src/open_r1"]

[tool.uv.sources]
transformers = { git = "https://github.com/huggingface/transformers.git", branch = "main" }
trl = { git = "https://github.com/huggingface/trl.git", branch = "grpo_vllm" }
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Use main branch here

Copy link
Author

@gerred gerred Jan 25, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

copy, I'm doing uv pip install -e . from that venv (which I'll document) in my trl clone after I uv sync, so that makes more sense anyway.

lighteval = { git = "https://github.com/huggingface/lighteval.git", rev = "4f381b352c0e467b5870a97d41cb66b487a2c503" }
torch = [
{ index = "pytorch-cu124" }
]

[[tool.uv.index]]
name = "pytorch-cu124"
url = "https://download.pytorch.org/whl/cu124"
explicit = true

[tool.isort]
default_section = "FIRSTPARTY"
ensure_newline_before_comments = true
force_grid_wrap = 0
include_trailing_comma = true
known_first_party = ["open_r1"]
known_third_party = [
"transformers",
"datasets",
"fugashi",
"git",
"h5py",
"matplotlib",
"nltk",
"numpy",
"packaging",
"pandas",
"psutil",
"pytest",
"rouge_score",
"sacrebleu",
"seqeval",
"sklearn",
"streamlit",
"torch",
"tqdm"
]
line_length = 119
lines_after_imports = 2
multi_line_output = 3
use_parentheses = true

[tool.flake8]
ignore = ["E203", "E501", "E741", "W503", "W605"]
max-line-length = 119
per-file-ignores = { "__init__.py" = ["F401"] }
41 changes: 0 additions & 41 deletions setup.cfg

This file was deleted.

135 changes: 0 additions & 135 deletions setup.py

This file was deleted.

Loading
Loading