Skip to content

Commit 3c164a9

Browse files
committed
fix: add transformers, torch, accelerate to [evaluator] extra
1 parent 1b8d629 commit 3c164a9

2 files changed

Lines changed: 14 additions & 2 deletions

File tree

pyproject.toml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[project]
22
name = "xfmr-zem"
3-
version = "0.3.2"
3+
version = "0.3.3"
44
description = "Zem: Unified Data Pipeline Framework (ZenML + NeMo Curator + DataJuicer) for multi-domain processing"
55
readme = "README.md"
66
requires-python = ">=3.10,<3.13"
@@ -119,6 +119,10 @@ voice = [
119119
# ── LLM / Evaluation ──────────────────────────────────────────────────────────
120120
evaluator = [
121121
"opik>=1.10.9",
122+
"transformers>=4.40.0",
123+
"torch>=2.1.0",
124+
"accelerate>=0.25.0",
125+
"litellm>=1.0.0",
122126
]
123127

124128
# ── Web UI ─────────────────────────────────────────────────────────────────────

uv.lock

Lines changed: 9 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)