Skip to content

Commit 9c14fb3

Browse files
authored
fix: HuggingFace image_uri generation for inference (#2730)
Test passed locally. Phillip also tested and worked for him.
1 parent 6250f63 commit 9c14fb3

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

src/sagemaker/huggingface/model.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -293,10 +293,6 @@ def serving_image_uri(self, region_name, instance_type, accelerator_type=None):
293293
str: The appropriate image URI based on the given parameters.
294294
295295
"""
296-
if image_uris._processor(instance_type, ["cpu", "gpu"]) == "gpu":
297-
container_version = "cu110-ubuntu18.04"
298-
else:
299-
container_version = "ubuntu18.04"
300296
if self.tensorflow_version is not None: # pylint: disable=no-member
301297
base_framework_version = (
302298
f"tensorflow{self.tensorflow_version}" # pylint: disable=no-member
@@ -312,5 +308,4 @@ def serving_image_uri(self, region_name, instance_type, accelerator_type=None):
312308
accelerator_type=accelerator_type,
313309
image_scope="inference",
314310
base_framework_version=base_framework_version,
315-
container_version=container_version,
316311
)

0 commit comments

Comments
 (0)