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