Skip to content

Commit 81e0d5a

Browse files
committed
update entry_point
1 parent b9d6003 commit 81e0d5a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/huggingface/estimator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,7 @@ def create_model(
250250
entry_point (str): Path (absolute or relative) to the local Python source file which
251251
should be executed as the entry point to training. If ``source_dir`` is specified,
252252
then ``entry_point`` must point to a file located at the root of ``source_dir``.
253-
If not specified, the training entry point is used.
253+
Defaults to `None`.
254254
source_dir (str): Path (absolute or relative) to a directory with any other serving
255255
source code dependencies aside from the entry point file.
256256
If not specified, the model source directory from training is used.
@@ -272,7 +272,7 @@ def create_model(
272272
return HuggingFaceModel(
273273
role or self.role,
274274
model_data=self.model_data,
275-
entry_point=entry_point or None,
275+
entry_point=entry_point,
276276
transformers_version=self.framework_version,
277277
tensorflow_version=self.tensorflow_version,
278278
pytorch_version=self.pytorch_version,

0 commit comments

Comments
 (0)