chore(dependencies): add neuronx-distributed for training extra

huggingface · Jan 29, 2025 · 43ad4be · 43ad4be
1 parent fa6df94
commit 43ad4be
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 = {