Kubeflow Trainer Official Release v2.1.0

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

Author: andreyvelich

api/python_api/kubeflow_trainer_api/__init__.py

@@ -20,7 +20,7 @@ name: kubeflow-trainer
 
 description: A Helm chart for deploying Kubeflow Trainer on Kubernetes.
 
-version: 2.1.0-rc.1
+version: 2.1.0
 
 type: application
 

charts/kubeflow-trainer/Chart.yaml

@@ -27,7 +27,7 @@ source "${SCRIPT_DIR}/../scripts/container-runtime.sh"
 setup_container_runtime
 
 # TODO (andreyvelich): Read this data from the global VERSION file.
-API_VERSION="2.1.0rc1"
+API_VERSION="2.1.0"
 API_OUTPUT_PATH="api/python_api"
 PKG_ROOT="${API_OUTPUT_PATH}/kubeflow_trainer_api"
 

hack/python-api/gen-api.sh

@@ -27,7 +27,7 @@ spec:
                     image: ghcr.io/kubeflow/trainer/dataset-initializer
                     env:
                       - name: CACHE_IMAGE
-                        value: "ghcr.io/kubeflow/trainer/data-cache:latest"
+                        value: "ghcr.io/kubeflow/trainer/data-cache:v2.1.0"
                       - name: TRAIN_JOB_NAME
                         valueFrom:
                           fieldRef:

manifests/base/runtimes/data-cache/torch_distributed_with_cache.yaml

@@ -12,7 +12,7 @@ resources:
 # Add data cache image tags
 images:
   - name: ghcr.io/kubeflow/trainer/dataset-initializer
-    newTag: latest
+    newTag: v2.1.0
 
 # Patch namespace-rbac resources to install in default namespace
 patches:

manifests/overlays/data-cache/kustomization.yaml

@@ -15,7 +15,7 @@ resources:
 # Update the Kubeflow Trainer controller manager image tag.
 images:
   - name: ghcr.io/kubeflow/trainer/trainer-controller-manager
-    newTag: v2.1.0-rc.1
+    newTag: v2.1.0
 
 # Secret for the Kubeflow Training webhook.
 secretGenerator:

manifests/overlays/manager/kustomization.yaml

@@ -6,12 +6,12 @@ resources:
 # Update the Kubeflow LLM Trainer image tag.
 images:
   - name: ghcr.io/kubeflow/trainer/torchtune-trainer
-    newTag: v2.1.0-rc.1
+    newTag: v2.1.0
   - name: ghcr.io/kubeflow/trainer/dataset-initializer
-    newTag: v2.1.0-rc.1
+    newTag: v2.1.0
   - name: ghcr.io/kubeflow/trainer/model-initializer
-    newTag: v2.1.0-rc.1
+    newTag: v2.1.0
   - name: ghcr.io/kubeflow/trainer/mlx-runtime
-    newTag: v2.1.0-rc.1
+    newTag: v2.1.0
   - name: ghcr.io/kubeflow/trainer/deepspeed-runtime
-    newTag: v2.1.0-rc.1
+    newTag: v2.1.0