From 1bb35da60fbe49f0b2f6ccfbfd991927e2baaedf Mon Sep 17 00:00:00 2001 From: Gavrish Prabhu Date: Fri, 29 Nov 2024 23:22:55 +0530 Subject: [PATCH] Pin accelerate package version in trainer (#2340) * Pin accelerate package version in trainer Signed-off-by: Gavrish Prabhu * include new line to pass pre-commit hook Signed-off-by: Gavrish Prabhu --------- Signed-off-by: Gavrish Prabhu Signed-off-by: sailesh duddupudi --- sdk/python/kubeflow/trainer/requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/sdk/python/kubeflow/trainer/requirements.txt b/sdk/python/kubeflow/trainer/requirements.txt index c2d306bbfb..59171ab910 100644 --- 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