renaming container image names

kubeflow · Jan 12, 2024 · cfcfaa7 · cfcfaa7
1 parent c4c236c
commit cfcfaa7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"