Skip to content

Commit 494d2c2

Browse files
dwarezalvarobartt
andcommitted
add: text-ranking to TEI tasks
Co-authored-by: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com>
1 parent 103ebce commit 494d2c2

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)