try with ModelReference first then with Model type

Author: Malav Shastri

src/sagemaker/jumpstart/accessors.py

@@ -301,8 +301,8 @@ def get_model_specs(
 
             except Exception as ex:
                 logging.info(
-                    "Recieved exeption while calling APIs for ContentType Model, \
-                        retrying with ContentType ModelReference: "
+                    "Recieved exeption while calling APIs for ContentType ModelReference, \
+                        retrying with ContentType Model: "
                     + str(ex)
                 )
                 hub_model_arn = construct_hub_model_arn_from_inputs(

src/sagemaker/jumpstart/hub/hub.py

@@ -276,37 +276,37 @@ def describe_model(
         try:
             model_version = get_hub_model_version(
                 hub_model_name=model_name,
-                hub_model_type=HubContentType.MODEL.value,
+                hub_model_type=HubContentType.MODEL_REFERENCE.value,
                 hub_name=self.hub_name,
                 sagemaker_session=self._sagemaker_session,
                 hub_model_version=model_version,
             )
 
             hub_content_description: Dict[str, Any] = self._sagemaker_session.describe_hub_content(
-                hub_name=self.hub_name if not hub_name else hub_name,
+                hub_name=self.hub_name,
                 hub_content_name=model_name,
                 hub_content_version=model_version,
-                hub_content_type=HubContentType.MODEL.value,
+                hub_content_type=HubContentType.MODEL_REFERENCE.value,
             )
 
         except Exception as ex:
             logging.info(
-                "Recieved exeption while calling APIs for ContentType Model, retrying with ContentType ModelReference: "
+                "Recieved exeption while calling APIs for ContentType ModelReference, retrying with ContentType Model: "
                 + str(ex)
             )
             model_version = get_hub_model_version(
                 hub_model_name=model_name,
-                hub_model_type=HubContentType.MODEL_REFERENCE.value,
+                hub_model_type=HubContentType.MODEL.value,
                 hub_name=self.hub_name,
                 sagemaker_session=self._sagemaker_session,
                 hub_model_version=model_version,
             )
 
             hub_content_description: Dict[str, Any] = self._sagemaker_session.describe_hub_content(
-                hub_name=self.hub_name,
+                hub_name=self.hub_name if not hub_name else hub_name,
                 hub_content_name=model_name,
                 hub_content_version=model_version,
-                hub_content_type=HubContentType.MODEL_REFERENCE.value,
+                hub_content_type=HubContentType.MODEL.value,
             )
 
         return DescribeHubContentResponse(hub_content_description)