Refactoring

Author: Jonathan Makunga

src/sagemaker/serve/builder/jumpstart_builder.py

@@ -282,8 +282,8 @@ def _tune_for_js(self, max_tuning_duration: int = 1800):
         ):
             admissible_tensor_parallel_degrees = [1]
             logger.warning(
-                "Sharded across multiple GPUs is not supported for this model."
-                "\nModel can only be sharded across [1] GPU"
+                "Sharded across multiple GPUs is not supported for this model. "
+                "Model can only be sharded across [1] GPU"
             )
 
         benchmark_results = {}

src/sagemaker/serve/utils/tuning.py

@@ -317,6 +317,9 @@ def sharded_supported(model_id: str, config_dict: dict) -> bool:
     """Check if sharded is supported for this ``Model``"""
     model_type = config_dict.get("model_type", None)
 
+    if model_type is None:
+        return False
+
     if model_id.startswith("facebook/galactica"):
         return True