Skip to content

Commit 4a947e6

Browse files
jyang0Jeffrey Yang
andauthored
update model version to 2.0 (#13216)
* update model version to 2.0 * Update supervised_example.py * Update supervised_advanced_example.py --------- Co-authored-by: Jeffrey Yang <jfyang@google.com>
1 parent 5da9655 commit 4a947e6

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

generative_ai/model_tuning/supervised_advanced_example.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -40,14 +40,16 @@ def gemini_tuning_advanced() -> sft.SupervisedTuningJob:
4040
# vertexai.init(encryption_spec_key_name="your-kms-key")
4141

4242
sft_tuning_job = sft.train(
43-
source_model="gemini-1.5-pro-002",
43+
source_model="gemini-2.0-flash-001",
44+
# 1.5 and 2.0 models use the same JSONL format
4445
train_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_train_data.jsonl",
4546
# The following parameters are optional
4647
validation_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_validation_data.jsonl",
47-
epochs=4,
48-
adapter_size=4,
49-
learning_rate_multiplier=1.0,
50-
tuned_model_display_name="tuned_gemini_1_5_pro",
48+
tuned_model_display_name="tuned_gemini_2_0_flash",
49+
# Advanced use only below. It is recommended to use auto-selection and leave them unset
50+
# epochs=4,
51+
# adapter_size=4,
52+
# learning_rate_multiplier=1.0,
5153
)
5254

5355
# Polling for job completion

generative_ai/model_tuning/supervised_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,8 @@ def gemini_tuning_basic() -> sft.SupervisedTuningJob:
3232
vertexai.init(project=PROJECT_ID, location="us-central1")
3333

3434
sft_tuning_job = sft.train(
35-
source_model="gemini-1.5-pro-002",
35+
source_model="gemini-2.0-flash-001",
36+
# 1.5 and 2.0 models use the same JSONL format
3637
train_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_train_data.jsonl",
3738
)
3839

0 commit comments

Comments
 (0)