From 43ad4be151e4c9eab53d63baf1163b0784b404bb Mon Sep 17 00:00:00 2001
From: Alvaro Moran <alvaro.moran@huggingface.co>
Date: Wed, 29 Jan 2025 10:14:11 +0000
Subject: [PATCH] chore(dependencies): add neuronx-distributed for training
 extra

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 917174e7b..292cba063 100644
--- a/setup.py
+++ b/setup.py
@@ -52,6 +52,7 @@
 TRAINING_REQUIRES = [
     "trl == 0.11.4",
     "peft == 0.14.0",
+    "neuronx-distributed == 0.9.0",
 ]
 
 EXTRAS_REQUIRE = {