简介

Hugging Face 适用于 Google Cloud 的深度学习容器是一组 Docker 镜像，用于在 Google Cloud Vertex AI 和 Google Kubernetes Engine (GKE) 上训练和部署 Transformers、Sentence Transformers 和 Diffusers 模型。

Google-Cloud-Containers 存储库包含用于构建 Hugging Face 特定深度学习容器 (DLC) 的容器文件，以及在 Google Cloud 上训练和部署模型的示例。这些容器由 Hugging Face 和 Google Cloud 团队定期公开维护、更新和发布，所有 Google Cloud 客户都可以在 Google Cloud 的 Artifact Registry 中使用它们。为每个受支持的用例组合（训练、推理）、加速器类型（CPU、GPU、TPU）和框架（PyTorch、TGI、TEI）创建了容器。

< > 在 GitHub 上更新