Skip to content

Commit 7ee3425

Browse files
authored
Merge branch 'master' into P50369404
2 parents c6c4c9d + 9c14fb3 commit 7ee3425

File tree

2 files changed

+7
-8
lines changed

2 files changed

+7
-8
lines changed

ci-scripts/queue_build.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,9 @@
1313
from __future__ import absolute_import
1414

1515
import os
16+
import re
1617
import time
18+
1719
import boto3
1820

1921
account = boto3.client(
@@ -23,9 +25,11 @@
2325

2426

2527
def queue_build():
26-
build_id = os.environ.get("CODEBUILD_BUILD_ID", "CODEBUILD-BUILD-ID")
27-
source_version = os.environ.get("CODEBUILD_SOURCE_VERSION", "CODEBUILD-SOURCE-VERSION").replace(
28-
"/", "-"
28+
build_id = re.sub("[_/]", "-", os.environ.get("CODEBUILD_BUILD_ID", "CODEBUILD-BUILD-ID"))
29+
source_version = re.sub(
30+
"[_/]",
31+
"-",
32+
os.environ.get("CODEBUILD_SOURCE_VERSION", "CODEBUILD-SOURCE-VERSION"),
2933
)
3034
ticket_number = int(1000 * time.time())
3135
filename = "%s_%s_%s" % (ticket_number, build_id, source_version)

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)