Pin accelerate package version in trainer (kubeflow#2340)

* Pin accelerate package version in trainer Signed-off-by: Gavrish Prabhu <gavrish.prabhu@nutanix.com> * include new line to pass pre-commit hook Signed-off-by: Gavrish Prabhu <gavrish.prabhu@nutanix.com> --------- Signed-off-by: Gavrish Prabhu <gavrish.prabhu@nutanix.com> Signed-off-by: sailesh duddupudi <saileshradar@gmail.com>
saileshd1402 · Dec 2, 2024 · 1bb35da · 1bb35da
1 parent 618bf6e
commit 1bb35da
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sdk/python/kubeflow/trainer/requirements.txt b/sdk/python/kubeflow/trainer/requirements.txt
@@ -1,3 +1,4 @@
 peft==0.3.0
 datasets==2.15.0
 transformers==4.38.0
+accelerate==0.28.0