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"