Skip to content

Commit 01c8491

Browse files
committed
test entry point
1 parent 7a71337 commit 01c8491

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/huggingface/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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 self._model_entry_point(),
275+
entry_point=entry_point or None,
276276
transformers_version=self.framework_version,
277277
tensorflow_version=self.tensorflow_version,
278278
pytorch_version=self.pytorch_version,

0 commit comments

Comments
 (0)