fix: add accelerate to ml optional dependencies

Trainer from transformers requires accelerate>=1.1.0 at runtime. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-12 21:23:05 -07:00 · 2026-02-18 10:51:51 -05:00
parent fcfe6890f6
commit cda9364071
1 changed files with 1 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,6 +23,7 @@ ml = [
    "torch>=2.0.0",
    "transformers>=4.30.0",
    "datasets>=2.14.0",
+    "accelerate>=1.1.0",
 ]
 dev = [
    "mypy>=1.8.0",