Skip to content

Commit b261f98

Browse files
committed
update container version
1 parent e336e05 commit b261f98

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/huggingface/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ def serving_image_uri(self, region_name, instance_type, accelerator_type=None):
294294
if instance_type == "gpu":
295295
container_version = "cu110-ubuntu18.04"
296296
else:
297-
container_version = None
297+
container_version = "ubuntu18.04"
298298
if self.tensorflow_version is not None: # pylint: disable=no-member
299299
base_framework_version = (
300300
f"tensorflow{self.tensorflow_version}" # pylint: disable=no-member

0 commit comments

Comments
 (0)