Skip to content

Commit

Permalink
change OnnxRuntime engine to Rust for text embedding
Browse files Browse the repository at this point in the history
  • Loading branch information
sindhuvahinis committed Dec 3, 2024
1 parent b19c894 commit e536429
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/sagemaker/djl_inference/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -148,7 +148,7 @@ def _infer_engine(self) -> Optional[str]:
return self.engine

if self.task == "text-embedding":
return "OnnxRuntime"
return "Rust"
return "Python"

def _infer_image_uri(self):
Expand Down

0 comments on commit e536429

Please sign in to comment.