We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 43ce1ba commit 6211227Copy full SHA for 6211227
src/sagemaker/serve/builder/transformers_builder.py
@@ -100,11 +100,9 @@ def _create_transformers_model(self) -> Type[Model]:
100
if hf_model_md is None:
101
raise ValueError("Could not fetch HF metadata")
102
103
- model_task = None
104
- if self.model_metadata:
+ model_task = hf_model_md.get("pipeline_tag")
+ if self.model_metadata.get("HF_TASK") is not None:
105
model_task = self.model_metadata.get("HF_TASK")
106
- else:
107
- model_task = hf_model_md.get("pipeline_tag")
108
109
if model_task == "sentence-similarity" and not self.image_uri:
110
self.image_uri = get_huggingface_llm_image_uri(
0 commit comments