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

huggingface · Jan 29, 2025 · 544da8d · 544da8d
1 parent e07a99c
commit 544da8d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -53,6 +53,7 @@
 TRAINING_REQUIRES = [
     "trl == 0.11.4",
     "peft == 0.14.0",
+    "neuronx-distributed == 0.9.0",
 ]
 
 EXTRAS_REQUIRE = {