From e1d6922edcd155545c7c0d44977f1475d82a0ee2 Mon Sep 17 00:00:00 2001
From: Justin Bollinger <justin.bollinger@trustedsec.com>
Date: Wed, 18 Feb 2026 10:51:51 -0500
Subject: [PATCH] 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>
---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index ed0b5b6..63185be 100644
--- 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",