Skip to content

Commit ed73c0d

Browse files
dwarezalvarobartt
andauthored
add: text-ranking to TEI tasks
Co-authored-by: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com>
1 parent 496b67c commit ed73c0d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

sagemaker-serve/src/sagemaker/serve/model_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2616,7 +2616,7 @@ def _build_single_modelbuilder(
26162616
elif model_task in OMNI_TASKS:
26172617
self.built_model = self._build_for_vllm_omni()
26182618
return self.built_model
2619-
elif model_task in ["sentence-similarity", "feature-extraction"]:
2619+
elif model_task in ["sentence-similarity", "feature-extraction", "text-ranking"]:
26202620
self.built_model = self._build_for_tei()
26212621
return self.built_model
26222622
else:

0 commit comments

Comments
 (0)