Kubeflow Trainer Official Release v2.0.1

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

Author: andreyvelich

api/python_api/kubeflow_trainer_api/__init__.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 
 
-__version__ = "2.0.0"
+__version__ = "2.0.1"

hack/python-api/gen-api.sh

@@ -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.0.0"
+API_VERSION="2.0.1"
 API_OUTPUT_PATH="api/python_api"
 PKG_ROOT="${API_OUTPUT_PATH}/kubeflow_trainer_api"
 

manifests/overlays/manager/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.0.0
+    newTag: v2.0.1
 
 # Secret for the Kubeflow Training webhook.
 secretGenerator:

manifests/overlays/runtimes/kustomization.yaml

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