Skip to content

Commit 4135331

Browse files
yishan-puYishan Pu
andauthored
feat(genai): update tuning gemini 2.5 flash base model name (#13439)
Co-authored-by: Yishan Pu <[email protected]>
1 parent c79f7d5 commit 4135331

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

genai/tuning/tuning_job_create.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ def create_tuning_job() -> str:
2323
client = genai.Client(http_options=HttpOptions(api_version="v1"))
2424

2525
tuning_job = client.tunings.tune(
26-
base_model="gemini-2.0-flash-lite-001",
27-
training_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini-2_0/text/sft_train_data.jsonl",
26+
base_model="gemini-2.5-flash",
27+
training_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini/text/sft_train_data.jsonl",
2828
config=CreateTuningJobConfig(
2929
tuned_model_display_name="Example tuning job",
3030
),

genai/tuning/tuning_with_checkpoints_create.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ def create_with_checkpoints() -> str:
2323
client = genai.Client(http_options=HttpOptions(api_version="v1"))
2424

2525
tuning_job = client.tunings.tune(
26-
base_model="gemini-2.0-flash-lite-001",
27-
training_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini-2_0/text/sft_train_data.jsonl",
26+
base_model="gemini-2.5-flash",
27+
training_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini/text/sft_train_data.jsonl",
2828
config=CreateTuningJobConfig(
2929
tuned_model_display_name="Example tuning job",
3030
# Set to True to disable tuning intermediate checkpoints. Default is False.

0 commit comments

Comments
 (0)