Skip to content

Commit a907ca4

Browse files
rlakhtakiaRadhika Lakhtakia
authored andcommitted
Add kaggle credentials to pytorch offering base job yamls
1 parent bf7e7f5 commit a907ca4

File tree

6 files changed

+33
-10
lines changed

6 files changed

+33
-10
lines changed

serving-catalog/core/deployment/jetstream/base/job.yaml

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,3 @@ spec:
1212
containers:
1313
- name: inference-checkpoint
1414
image: us-docker.pkg.dev/cloud-tpu-images/inference/inference-checkpoint:v0.2.2
15-
volumeMounts:
16-
- mountPath: "/kaggle/"
17-
name: kaggle-credentials
18-
readOnly: true
19-
volumes:
20-
- name: kaggle-credentials
21-
secret:
22-
defaultMode: 0400
23-
secretName: kaggle-secret

serving-catalog/core/deployment/jetstream/gemma-7b-it/base/job.patch.yaml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,3 +15,12 @@ spec:
1515
args:
1616
- -b=$BUCKET_NAME
1717
- -m=google/gemma/maxtext/7b-it/2
18+
volumeMounts:
19+
- mountPath: "/kaggle/"
20+
name: kaggle-credentials
21+
readOnly: true
22+
volumes:
23+
- name: kaggle-credentials
24+
secret:
25+
defaultMode: 0400
26+
secretName: kaggle-secret

serving-catalog/core/deployment/jetstream/llama3-8b/base/job.patch.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,15 @@ spec:
2424
- mountPath: "/huggingface/"
2525
name: huggingface-credentials
2626
readOnly: true
27+
- mountPath: "/kaggle/"
28+
name: kaggle-credentials
29+
readOnly: true
2730
volumes:
2831
- name: huggingface-credentials
2932
secret:
3033
defaultMode: 0400
3134
secretName: huggingface-secret
35+
- name: kaggle-credentials
36+
secret:
37+
defaultMode: 0400
38+
secretName: kaggle-secret

serving-catalog/core/deployment/jetstream/maxtext/llama3-8b/base/job.patch.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ spec:
1818
- --inference_server=jetstream-maxtext
1919
- --model_path=Llama3.1-8B-Instruct
2020
- --output_directory=gs://$BUCKET_NAME/maxtext/llama3-8b/final/bf16/
21-
- --model_name=llama-3
21+
- --model_name=llama-3.1
2222
- --huggingface=False
2323
- --bucket_name=$BUCKET_NAME
2424
- --meta_url=$META_URL

serving-catalog/core/deployment/jetstream/pytorch/gemma-7b-it/base/job.patch.yaml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,3 +15,12 @@ spec:
1515
args:
1616
- -b=$BUCKET_NAME
1717
- -m=google/gemma/maxtext/7b-it/2
18+
volumeMounts:
19+
- mountPath: "/kaggle/"
20+
name: kaggle-credentials
21+
readOnly: true
22+
volumes:
23+
- name: kaggle-credentials
24+
secret:
25+
defaultMode: 0400
26+
secretName: kaggle-secret

serving-catalog/core/deployment/jetstream/pytorch/llama3-8b/base/job.patch.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,15 @@ spec:
2424
- mountPath: "/huggingface/"
2525
name: huggingface-credentials
2626
readOnly: true
27+
- mountPath: "/kaggle/"
28+
name: kaggle-credentials
29+
readOnly: true
2730
volumes:
2831
- name: huggingface-credentials
2932
secret:
3033
defaultMode: 0400
3134
secretName: huggingface-secret
35+
- name: kaggle-credentials
36+
secret:
37+
defaultMode: 0400
38+
secretName: kaggle-secret

0 commit comments

Comments
 (0)