Skip to content

Commit 7210d2f

Browse files
committed
added to do
1 parent 02445ba commit 7210d2f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/sagemaker/huggingface/model.py

+3
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,9 @@ def __init__(
174174

175175
self.model_server_workers = model_server_workers
176176

177+
# TODO: Remove the following function once botocore adds hugginface to the list of valid neo compilable frameworks.
178+
# Ideally with inferentia framewrok, call to .compile( ... ) method will create the image_uri.
179+
# currently, call to .compile is causing `ValidationException`
177180
def deploy(
178181
self,
179182
initial_instance_count=None,

0 commit comments

Comments
 (0)