add cpu back

pagezyhf · pagezyhf · commit 47b27f7357f3 · 2025-10-16T11:30:53.000+02:00
diff --git a/src/sagemaker/image_uri_config/huggingface.json b/src/sagemaker/image_uri_config/huggingface.json
@@ -2181,7 +2181,8 @@
                     },
                     "repository": "huggingface-pytorch-inference",
                     "container_version": {
-                        "gpu": "cu124-ubuntu22.04"
+                        "gpu": "cu124-ubuntu22.04",
+                        "cpu": "ubuntu22.04"
                     }
                 }
             }

Original file line number	Diff line number	Diff line change
`@@ -2181,7 +2181,8 @@`
`2181`	`2181`	`},`
`2182`	`2182`	`"repository": "huggingface-pytorch-inference",`
`2183`	`2183`	`"container_version": {`
`2184`		`- "gpu": "cu124-ubuntu22.04"`
	`2184`	`+ "gpu": "cu124-ubuntu22.04",`
	`2185`	`+ "cpu": "ubuntu22.04"`
`2185`	`2186`	`}`
`2186`	`2187`	`}`
`2187`	`2188`	`}`