From 7fe0b656061e3dc33dac31f038f49503cd53734b Mon Sep 17 00:00:00 2001 From: Ryan Cook Date: Wed, 16 Oct 2024 11:08:09 -0400 Subject: [PATCH] rhelai 1.2 ga Signed-off-by: Ryan Cook --- pipeline.yaml | 4 ++-- standalone/standalone.py | 2 +- standalone/standalone.tpl | 2 +- training/components.py | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/pipeline.yaml b/pipeline.yaml index 704c1200..fe6e46d7 100644 --- a/pipeline.yaml +++ b/pipeline.yaml @@ -783,7 +783,7 @@ deploymentSpec: \ = f\"train-{phase_name}-{name_suffix.rstrip('-sdg')}\"\n if phase_name\ \ == \"first\":\n path_to_model = \"/input_model/model\"\n elif\ \ phase_name == \"second\":\n path_to_model = list_phase1_final_model()\n\ - \ image = \"registry.stage.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2\"\ + \ image = \"registry.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2\"\ \n\n manifest = inspect.cleandoc(\n f\"\"\"\n apiVersion:\ \ kubeflow.org/v1\n kind: PyTorchJob\n metadata:\n \ \ name: {name}\n spec:\n nprocPerNode: \\\\\"{nproc_per_node}\\\ @@ -903,7 +903,7 @@ deploymentSpec: \ = f\"train-{phase_name}-{name_suffix.rstrip('-sdg')}\"\n if phase_name\ \ == \"first\":\n path_to_model = \"/input_model/model\"\n elif\ \ phase_name == \"second\":\n path_to_model = list_phase1_final_model()\n\ - \ image = \"registry.stage.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2\"\ + \ image = \"registry.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2\"\ \n\n manifest = inspect.cleandoc(\n f\"\"\"\n apiVersion:\ \ kubeflow.org/v1\n kind: PyTorchJob\n metadata:\n \ \ name: {name}\n spec:\n nprocPerNode: \\\\\"{nproc_per_node}\\\ diff --git a/standalone/standalone.py b/standalone/standalone.py index 99d48b1e..8cbfd0d0 100755 --- a/standalone/standalone.py +++ b/standalone/standalone.py @@ -47,7 +47,7 @@ K8S_NAME = "kfp-model-server" TOOLBOX_IMAGE = "registry.access.redhat.com/ubi9/toolbox" DS_IMAGE = "quay.io/opendatahub/workbench-images:jupyter-datascience-ubi9-python-3.11-20241004-609ffb8" -RHELAI_IMAGE = "registry.stage.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2" +RHELAI_IMAGE = "registry.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2" DATA_PVC_NAME = "data" DATA_PVC_MOUNT_PATH = "/data" DATA_PVC_MODEL_PATH = path.join(DATA_PVC_MOUNT_PATH, "model") diff --git a/standalone/standalone.tpl b/standalone/standalone.tpl index 832c88e9..5e6653b4 100755 --- a/standalone/standalone.tpl +++ b/standalone/standalone.tpl @@ -47,7 +47,7 @@ DEFAULT_REPO_URL = "https://github.com/instructlab/taxonomy.git" K8S_NAME = "kfp-model-server" TOOLBOX_IMAGE = "registry.access.redhat.com/ubi9/toolbox" DS_IMAGE = "quay.io/opendatahub/workbench-images:jupyter-datascience-ubi9-python-3.11-20241004-609ffb8" -RHELAI_IMAGE = "registry.stage.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2" +RHELAI_IMAGE = "registry.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2" DATA_PVC_NAME = "data" DATA_PVC_MOUNT_PATH = "/data" DATA_PVC_MODEL_PATH = path.join(DATA_PVC_MOUNT_PATH, "model") diff --git a/training/components.py b/training/components.py index e4f7a78f..79d71422 100644 --- a/training/components.py +++ b/training/components.py @@ -109,7 +109,7 @@ def list_phase1_final_model(): path_to_model = "/input_model/model" elif phase_name == "second": path_to_model = list_phase1_final_model() - image = "registry.stage.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2" + image = "registry.redhat.io/rhelai1/instructlab-nvidia-rhel9:1.2" manifest = inspect.cleandoc( f"""