tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning¶

Example Airflow DAG for Google Vertex AI Generative Model Tuning Tasks.

Attributes¶

`PROJECT_ID`
`DAG_ID`
`REGION`
`SOURCE_MODEL`
`TRAIN_DATASET`
`TUNED_MODEL_DISPLAY_NAME`
`sft_train_task`
`test_run`

Module Contents¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.PROJECT_ID[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.DAG_ID = 'vertex_ai_generative_model_tuning_dag'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.REGION = 'us-central1'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.SOURCE_MODEL = 'gemini-1.0-pro-002'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.TRAIN_DATASET = 'gs://cloud-samples-data/ai-platform/generative_ai/sft_train_data.jsonl'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.TUNED_MODEL_DISPLAY_NAME = 'my_tuned_gemini_model'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.sft_train_task[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.test_run[source]¶

Was this entry helpful?