-
Notifications
You must be signed in to change notification settings - Fork 22
Expand file tree
/
Copy pathpyproject.toml
More file actions
42 lines (38 loc) · 1.04 KB
/
Copy pathpyproject.toml
File metadata and controls
42 lines (38 loc) · 1.04 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "meshy"
version = "0.1.0"
description = "A minimal GRPO reinforcement-learning trainer over torchtitan + SGLang."
requires-python = ">=3.12"
# NOTE: torch, torchtitan and sglang are installed manually (see README.md) —
# torch/torchtitan come from the PyTorch nightly index and are deliberately NOT
# listed here so that `pip install -e .` does not pull stable wheels and clobber
# the nightly build. sglang is excluded for the same reason: it would drag its
# own pinned stable torch in behind it.
dependencies = [
"transformers",
"datasets",
"fastapi",
"httpx",
"cloudpickle",
"loguru",
"tensordict",
"pybase64",
"huggingface_hub",
"sympy",
"pylatexenc",
"uvicorn",
"pyzmq",
"hydra-core",
"numpy",
"msgspec",
"psutil",
"omegaconf",
"prometheus_client>=0.20.0",
]
[project.optional-dependencies]
wandb = ["wandb"]
[tool.setuptools.packages.find]
include = ["meshy*"]