feat: allowing users to specify the version id of the Model Garden model

Google APIs · copybara-github · commit 7dfaf2724f95 · 2025-03-04T21:03:27.000-08:00
feat: allowing users to choose whether to use the hf model cache

PiperOrigin-RevId: 733567956
diff --git a/google/cloud/aiplatform/v1/model.proto b/google/cloud/aiplatform/v1/model.proto
@@ -493,6 +493,12 @@ message LargeModelReference {
 message ModelGardenSource {
   // Required. The model garden source model resource name.
   string public_model_name = 1 [(google.api.field_behavior) = REQUIRED];
+
+  // Optional. The model garden source model version ID.
+  string version_id = 3 [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Whether to avoid pulling the model from the HF cache.
+  bool skip_hf_model_cache = 4 [(google.api.field_behavior) = OPTIONAL];
 }
 
 // Contains information about the source of the models generated from Generative
