Update model.py

njbrake · web-flow · commit fec3b8800e09 · 2024-05-22T15:02:32.000-04:00
Allow for the container to be passed the image_config
diff --git a/src/sagemaker/huggingface/model.py b/src/sagemaker/huggingface/model.py
@@ -523,6 +523,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,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -523,6 +523,7 @@ def prepare_container_def(`
`523`	`523`	`deploy_image,`
`524`	`524`	`self.repacked_model_data or self.model_data,`
`525`	`525`	`deploy_env,`
	`526`	`+ image_config=self.image_config,`
`526`	`527`	`accept_eula=accept_eula,`
`527`	`528`	`)`
`528`	`529`