From ac2c2ff7045d58f456d837c417c50d27e23c33c4 Mon Sep 17 00:00:00 2001 From: Mofi Rahman Date: Thu, 11 Jul 2024 12:42:27 -0400 Subject: [PATCH] update image url for gemma finetune yaml (#729) * Update finetune.yaml --- .../genAI-LLM/finetuning-gemma-2b-on-l4/finetune.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorials-and-examples/genAI-LLM/finetuning-gemma-2b-on-l4/finetune.yaml b/tutorials-and-examples/genAI-LLM/finetuning-gemma-2b-on-l4/finetune.yaml index 908fe4b09..98f71b92d 100644 --- a/tutorials-and-examples/genAI-LLM/finetuning-gemma-2b-on-l4/finetune.yaml +++ b/tutorials-and-examples/genAI-LLM/finetuning-gemma-2b-on-l4/finetune.yaml @@ -15,7 +15,7 @@ spec: terminationGracePeriodSeconds: 600 containers: - name: finetuner - image: + image: $IMAGE_URL resources: limits: nvidia.com/gpu: "8" @@ -23,7 +23,7 @@ spec: - name: MODEL_NAME value: "google/gemma-2b" - name: NEW_MODEL - value: "" + value: "gemma-2b-sql-finetuned" - name: LORA_R value: "8" - name: LORA_ALPHA