Skip to content

Commit

Permalink
update image url for gemma finetune yaml (GoogleCloudPlatform#729)
Browse files Browse the repository at this point in the history
* Update finetune.yaml
  • Loading branch information
moficodes committed Jul 11, 2024
1 parent 2bfbcd7 commit ac2c2ff
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -15,15 +15,15 @@ spec:
terminationGracePeriodSeconds: 600
containers:
- name: finetuner
image: <YOUR_IMAGE>
image: $IMAGE_URL
resources:
limits:
nvidia.com/gpu: "8"
env:
- name: MODEL_NAME
value: "google/gemma-2b"
- name: NEW_MODEL
value: "<FINETUNED_MODEL_ID>"
value: "gemma-2b-sql-finetuned"
- name: LORA_R
value: "8"
- name: LORA_ALPHA
Expand Down

0 comments on commit ac2c2ff

Please sign in to comment.