From cfcfaa79c2caa11271820f8a9aaed0756bff8c07 Mon Sep 17 00:00:00 2001 From: deepanker13 Date: Fri, 12 Jan 2024 19:27:51 +0530 Subject: [PATCH] renaming container image names --- sdk/python/kubeflow/training/constants/constants.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/python/kubeflow/training/constants/constants.py b/sdk/python/kubeflow/training/constants/constants.py index f2df70b82f..83d3703ee8 100644 --- a/sdk/python/kubeflow/training/constants/constants.py +++ b/sdk/python/kubeflow/training/constants/constants.py @@ -88,8 +88,8 @@ PYTORCHJOB_REPLICA_TYPES = (REPLICA_TYPE_MASTER.lower(), REPLICA_TYPE_WORKER.lower()) PYTORCHJOB_BASE_IMAGE = "docker.io/pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime" STORAGE_CONTAINER = "storage-initializer" -STORAGE_CONTAINER_IMAGE = "quay.io/deepanker_gupta/storage:v1" -TRAINER_TRANSFORMER_IMAGE = "quay.io/deepanker_gupta/trainer:v1" +STORAGE_CONTAINER_IMAGE = "docker.io/kubeflow/storage-initializer" +TRAINER_TRANSFORMER_IMAGE = "docker.io/kubeflow/trainer-huggingface" TRAINER_PVC_NAME = "storage-initializer" TRAINER_PV = "storage-pv"