Update model.py (#4700)

njbrake · sage-maker · web-flow · commit 7b1a7ad30b85 · 2024-08-12T09:01:40.000-07:00
Allow for the container to be passed the image_config

Co-authored-by: sage-maker &lt;parknate@amazon.com&gt;
diff --git a/src/sagemaker/huggingface/model.py b/src/sagemaker/huggingface/model.py
@@ -533,6 +533,7 @@ def prepare_container_def(
             deploy_image,
             self.repacked_model_data or self.model_data,
             deploy_env,
+            image_config=self.image_config,
             accept_eula=accept_eula,
             model_reference_arn=model_reference_arn,
         )

Original file line number	Diff line number	Diff line change
`@@ -533,6 +533,7 @@ def prepare_container_def(`
`533`	`533`	`deploy_image,`
`534`	`534`	`self.repacked_model_data or self.model_data,`
`535`	`535`	`deploy_env,`
	`536`	`+ image_config=self.image_config,`
`536`	`537`	`accept_eula=accept_eula,`
`537`	`538`	`model_reference_arn=model_reference_arn,`
`538`	`539`	`)`