We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aceb8e0 commit a0a92bdCopy full SHA for a0a92bd
src/sagemaker/pytorch/estimator.py
@@ -175,7 +175,7 @@ def create_model(
175
code_location=self.code_location,
176
py_version=self.py_version,
177
framework_version=self.framework_version,
178
- image=self.image_name,
+ image=kwargs["image"] if "image" in kwargs else self.image_name,
179
model_server_workers=model_server_workers,
180
sagemaker_session=self.sagemaker_session,
181
vpc_config=self.get_vpc_config(vpc_config_override),
0 commit comments