简介

用于 Google Cloud 的 Hugging Face 深度学习容器 是一组 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 上