Sagemaker-huggingface 调用CreateModel操作时发生错误(ValidationException):未找到请求的图像

问题描述 投票:0回答:1

我正在尝试创建一个异步 sagemaker 端点,如此处所示 此处所示 但使用拥抱脸模型,因此我需要使用拥抱脸图像之一,为此我检查可用选项的官方列表 此处 但是当尝试基于此创建模型,我收到错误:

ClientError: An error occurred (ValidationException) when calling the CreateModel operation: Requested image 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training:1.9.1-transformers4.12-gpu-py38-cu111-ubuntu20.04 not found.

我测试了官方列表中的 3 个选项,但没有一个有效(甚至该页面中显示的示例也无效)。但是使用完全相同的代码,只需更改为 xgboost 及其相应的版本就可以了,所以我知道它与此图像有特定的关系,关于在哪里可以找到支持的 Huggingface 图像列表有什么想法或建议吗?

python amazon-web-services machine-learning amazon-sagemaker huggingface
1个回答
0
投票

它看起来像一个旧容器,可能不受支持,您可以尝试从此 URL 更改为最新的 Huggingface 容器图像,看看它是否有效

https://github.com/aws/deep-learning-containers/blob/master/available_images.md#huggingface-inference-containers

© www.soinside.com 2019 - 2024. All rights reserved.