Skip to content

Commit 17a0f32

Browse files
committed
Fix issue with entry point in serving.properties
1 parent ff8f223 commit 17a0f32

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/djl_inference/model.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -243,6 +243,7 @@ def prepare_container_def(
243243
else self.sagemaker_session.settings.local_download_dir
244244
)
245245
with _tmpdir(directory=local_download_dir) as tmp:
246+
logger.info(f"Using tmp dir {tmp}")
246247
if self.source_dir or self.entry_point:
247248
_create_or_update_code_dir(
248249
tmp,
@@ -284,7 +285,7 @@ def generate_serving_properties(self, serving_properties={}) -> Dict[str, str]:
284285
if self.tensor_parallel_degree:
285286
serving_properties["option.tensor_parallel_degree"] = self.tensor_parallel_degree
286287
if self.entry_point:
287-
serving_properties["entryPoint"] = self.entry_point
288+
serving_properties["option.entryPoint"] = self.entry_point
288289
if self.task:
289290
serving_properties["option.task"] = self.task
290291
if self.data_type:

0 commit comments

Comments
 (0)