File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/sagemaker/huggingface Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -250,7 +250,7 @@ def create_model(
250
250
entry_point (str): Path (absolute or relative) to the local Python source file which
251
251
should be executed as the entry point to training. If ``source_dir`` is specified,
252
252
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` .
254
254
source_dir (str): Path (absolute or relative) to a directory with any other serving
255
255
source code dependencies aside from the entry point file.
256
256
If not specified, the model source directory from training is used.
@@ -272,7 +272,7 @@ def create_model(
272
272
return HuggingFaceModel (
273
273
role or self .role ,
274
274
model_data = self .model_data ,
275
- entry_point = entry_point or None ,
275
+ entry_point = entry_point ,
276
276
transformers_version = self .framework_version ,
277
277
tensorflow_version = self .tensorflow_version ,
278
278
pytorch_version = self .pytorch_version ,
You can’t perform that action at this time.
0 commit comments