Pin accelerate package version in trainer (#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>
kubeflow · Nov 29, 2024 · cb92b36 · cb92b36
1 parent b7e4028
commit cb92b36
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