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

Google APIs · copybara-github · commit 7e795c44dae3 · 2025-03-05T08:48:15.000-08:00
feat: allowing users to choose whether to use the hf model cache

PiperOrigin-RevId: 733750388
diff --git a/google/cloud/aiplatform/v1beta1/model.proto b/google/cloud/aiplatform/v1beta1/model.proto
@@ -452,6 +452,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
