publish trainer hugging face image (#1985)

* publish trainer hugging face * renaming container image names
kubeflow · Jan 12, 2024 · 521cbed · 521cbed
1 parent 5a9b189
commit 521cbed
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/publish-core-images.yaml b/.github/workflows/publish-core-images.yaml
@@ -32,6 +32,6 @@ jobs:
             context: sdk/python/kubeflow/storage_initializer
             platforms: linux/amd64,linux/arm64
           - component-name: trainer-huggingface
-            dockerfile: sdk/python/kubeflow/trainer/hf_dockerfile
+            dockerfile: sdk/python/kubeflow/trainer/Dockerfile
             context: sdk/python/kubeflow/trainer
             platforms: linux/amd64,linux/arm64
diff --git a/sdk/python/kubeflow/trainer/hf_dockerfile → sdk/python/kubeflow/trainer/Dockerfile b/sdk/python/kubeflow/trainer/hf_dockerfile → sdk/python/kubeflow/trainer/Dockerfile
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"