NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove dependency on torch for default installation

Browse source
This commit is contained in:
dmahan93 2025-05-12 10:17:41 -05:00
parent 706097db21
commit 727c7ba640
5 changed files with 57 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

9
pyproject.toml
View file

@ -24,7 +24,6 @@ dependencies = [
"polars",
"aiofiles",
"jsonlines",
"torch",
"pydantic-cli",
"hf_transfer",
]
@ -41,6 +40,9 @@ atropos-dpo-gen = "atroposlib.cli.dpo:main"
all = [
"atroposlib[dev,examples]"
]
rewardfns = [
"torch"
]
dev = [
"pytest",
"pytest-asyncio",
@ -49,10 +51,11 @@ dev = [
"flake8",
"isort",
"mypy",
'rich',
"rich",
]
examples = [
"gradio"
"gradio",
"atroposlib[rewardfns]"
]
[build-system]