Skip to content

Commit b9d6003

Browse files
committed
add processor
1 parent b261f98 commit b9d6003

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
@@ -291,7 +291,7 @@ def serving_image_uri(self, region_name, instance_type, accelerator_type=None):
291291
str: The appropriate image URI based on the given parameters.
292292
293293
"""
294-
if instance_type == "gpu":
294+
if image_uris._processor(instance_type, ["cpu", "gpu"]) == "gpu":
295295
container_version = "cu110-ubuntu18.04"
296296
else:
297297
container_version = "ubuntu18.04"

0 commit comments

Comments
 (0)