fix: format of the input to GetPublisherModelRequest for Hugging Face models.

PiperOrigin-RevId: 738211073

Author: vertex-sdk-bot

tests/unit/vertexai/model_garden/test_model_garden.py

@@ -33,10 +33,16 @@
 _TEST_MODEL_FULL_RESOURCE_NAME = (
  "publishers/google/models/paligemma@paligemma-224-float32"
 )
+_TEST_HUGGING_FACE_MODEL_FULL_RESOURCE_NAME = (
+ "publishers/meta-llama/models/llama-3.3-70b-instruct"
+)
 _TEST_PUBLISHER_MODEL_NAME = "publishers/google/models/paligemma"
 _TEST_HUGGING_FACE_PUBLISHER_MODEL_NAME = "publishers/hf-google/models/gemma-2-2b"
 _TEST_MODEL_SIMPLIFIED_RESOURCE_NAME = "google/paligemma@paligemma-224-float32"
 _TEST_MODEL_HUGGING_FACE_ID = "meta-llama/Llama-3.3-70B-Instruct"
+_TEST_MODEL_HUGGING_FACE_RESOURCE_NAME = (
+ "publishers/hf-meta-llama/models/llama-3.3-70b-instruct"
+)
 # Note: The full resource name is in lower case.
 _TEST_MODEL_HUGGING_FACE_FULL_RESOURCE_NAME = (
  "publishers/hf-meta-llama/models/llama-3.3-70b-instruct@001"
@@ -120,6 +126,24 @@ def get_publisher_model_mock():
  )
  ),
  ),
+ types.PublisherModel(
+ name=_TEST_MODEL_HUGGING_FACE_RESOURCE_NAME,
+ supported_actions=types.PublisherModel.CallToAction(
+ multi_deploy_vertex=types.PublisherModel.CallToAction.DeployVertex(
+ multi_deploy_vertex=[
+ types.PublisherModel.CallToAction.Deploy(
+ dedicated_resources=types.DedicatedResources(
+ machine_spec=types.MachineSpec(
+ machine_type="g2-standard-16",
+ accelerator_type="NVIDIA_L4",
+ accelerator_count=1,
+ )
+ )
+ )
+ ]
+ )
+ ),
+ ),
  ]
  yield get_publisher_model_mock
 
@@ -758,6 +782,16 @@ def test_list_deploy_options(self, get_publisher_model_mock):
  )
  )
 
+ hf_model = model_garden.OpenModel(_TEST_MODEL_HUGGING_FACE_ID)
+ hf_model.list_deploy_options()
+ get_publisher_model_mock.assert_called_with(
+ types.GetPublisherModelRequest(
+ name=_TEST_HUGGING_FACE_MODEL_FULL_RESOURCE_NAME,
+ is_hugging_face_model=True,
+ include_equivalent_model_garden_model_deployment_configs=True,
+ )
+ )
+
  def test_list_deployable_models(self, list_publisher_models_mock):
  """Tests getting the supported deploy options for a model."""
  aiplatform.init(

vertexai/model_garden/_model_garden.py

@@ -147,8 +147,8 @@ def _reconcile_model_name(model_name: str) -> str:
  )
  else:
  return _get_publisher_model_resource_name(
- publisher="hf-" + simplified_name_match.group("publisher"),
- model=simplified_name_match.group("model") + "@001",
+ publisher=simplified_name_match.group("publisher"),
+ model=simplified_name_match.group("model"),
  )
  else:
  raise ValueError(f"`{model_name}` is not a valid Open Model name")
@@ -594,7 +594,7 @@ def list_deploy_options(
  """Lists the verified deploy options for the model."""
  request = types.GetPublisherModelRequest(
  name=self._publisher_model_name,
- is_hugging_face_model="@" not in self._publisher_model_name,
+ is_hugging_face_model=bool(self._is_hugging_face_model),
  include_equivalent_model_garden_model_deployment_configs=True,
  )
  response = self._us_central1_model_garden_client.get_publisher_model(request)