Skip to content

Commit 11b9cc6

Browse files
committed
feat(gke): pin transformer and dataset versions in hyperdisk example
1 parent 8183d57 commit 11b9cc6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/gke-managed-hyperdisk.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ deployment_groups:
146146
- bash
147147
- -c
148148
- |
149-
pip install transformers datasets
149+
pip install transformers==4.54.1 dataset==4.0.0
150150
python - <<EOF
151151
from datasets import load_dataset
152152
dataset = load_dataset("glue", "cola", cache_dir='/data/hyperdisk-balanced-pvc-0')
@@ -188,7 +188,7 @@ deployment_groups:
188188
- bash
189189
- -c
190190
- |
191-
pip install transformers datasets
191+
pip install transformers==4.54.1 dataset==4.0.0
192192
python - <<EOF
193193
from datasets import load_dataset
194194
dataset = load_dataset("glue", "cola", cache_dir='/data/hyperdisk-extreme-pvc-0')
@@ -230,7 +230,7 @@ deployment_groups:
230230
- bash
231231
- -c
232232
- |
233-
pip install transformers datasets
233+
pip install transformers==4.54.1 dataset==4.0.0
234234
python - <<EOF
235235
from datasets import load_dataset
236236
dataset = load_dataset("glue", "cola", cache_dir='/data/hyperdisk-throughput-pvc-0')

0 commit comments

Comments
 (0)