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