Skip to content

Commit e536429

Browse files
committed
change OnnxRuntime engine to Rust for text embedding
1 parent b19c894 commit e536429

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/djl_inference/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def _infer_engine(self) -> Optional[str]:
148148
return self.engine
149149

150150
if self.task == "text-embedding":
151-
return "OnnxRuntime"
151+
return "Rust"
152152
return "Python"
153153

154154
def _infer_image_uri(self):

0 commit comments

Comments
 (0)