Debugging issue

Author: samruds

src/sagemaker/serve/builder/model_builder.py

@@ -451,6 +451,7 @@ def _create_model(self):
         self._original_register = self.pysdk_model.register
         self.pysdk_model.register = self._model_builder_register_wrapper
         self.model_package = None
+        logger.info("Model %s registered", self.pysdk_model)
         return self.pysdk_model
 
     @_capture_telemetry("register")
@@ -565,6 +566,7 @@ def _build_for_torchserve(self) -> Type[Model]:
         )
 
         self._prepare_for_mode()
+        logger.info("Building for torchserve")
 
         return self._create_model()
 
@@ -625,9 +627,6 @@ def build(  # pylint: disable=R0911
         self._is_custom_image_uri = self.image_uri is not None
 
         if isinstance(self.model, str):
-            model_task = None
-            if self.model_metadata:
-                model_task = self.model_metadata.get("HF_TASK")
             if self._is_jumpstart_model_id():
                 return self._build_for_jumpstart()
             if self._is_djl():  # pylint: disable=R1705
@@ -636,20 +635,8 @@ def build(  # pylint: disable=R0911
                 hf_model_md = get_huggingface_model_metadata(
                     self.model, self.env_vars.get("HUGGING_FACE_HUB_TOKEN")
                 )
-
-                if model_task is None:
-                    model_task = hf_model_md.get("pipeline_tag")
-                if self.schema_builder is None and model_task is not None:
-                    self._schema_builder_init(model_task)
-                if model_task == "text-generation":  # pylint: disable=R1705
+                if hf_model_md.get("pipeline_tag") == "text-generation":  # pylint: disable=R1705
                     return self._build_for_tgi()
-                elif self._can_fit_on_single_gpu():
-                    return self._build_for_transformers()
-                elif (
-                    self.model in defaults.DEEPSPEED_RECOMMENDED_ARCHITECTURES
-                    or self.model in defaults.FASTER_TRANSFORMER_RECOMMENDED_ARCHITECTURES
-                ):
-                    return self._build_for_djl()
                 else:
                     return self._build_for_transformers()