From 5baddf5259ab75be6b6bdee9f847700e91defd38 Mon Sep 17 00:00:00 2001
From: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com>
Date: Fri, 29 Nov 2024 09:26:40 +0100
Subject: [PATCH] Add missing `google-cloud-storage` installation (#124)

---
 .../inference/cpu/2.2.2/transformers/4.41.1/py311/Dockerfile   | 3 +++
 .../inference/gpu/2.2.2/transformers/4.41.1/py311/Dockerfile   | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/containers/pytorch/inference/cpu/2.2.2/transformers/4.41.1/py311/Dockerfile b/containers/pytorch/inference/cpu/2.2.2/transformers/4.41.1/py311/Dockerfile
index 85f3538f..fdaff531 100644
--- a/containers/pytorch/inference/cpu/2.2.2/transformers/4.41.1/py311/Dockerfile
+++ b/containers/pytorch/inference/cpu/2.2.2/transformers/4.41.1/py311/Dockerfile
@@ -39,6 +39,9 @@ ARG HF_INFERENCE_TOOLKIT_VERSION=0.4.1
 ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
 RUN pip install "${HF_INFERENCE_TOOLKIT_URL}#egg=huggingface-inference-toolkit[torch,diffusers,st]"
 
+# Install google-cloud-storage required to download artifacts from Google Cloud Storage buckets
+RUN pip install --upgrade google-cloud-storage
+
 # copy entrypoint and change permissions
 COPY --chmod=0755 containers/pytorch/inference/cpu/2.2.2/transformers/4.41.1/py311/entrypoint.sh entrypoint.sh
 ENTRYPOINT ["bash", "-c", "./entrypoint.sh"]
diff --git a/containers/pytorch/inference/gpu/2.2.2/transformers/4.41.1/py311/Dockerfile b/containers/pytorch/inference/gpu/2.2.2/transformers/4.41.1/py311/Dockerfile
index 7ccec8b7..69047fc5 100644
--- a/containers/pytorch/inference/gpu/2.2.2/transformers/4.41.1/py311/Dockerfile
+++ b/containers/pytorch/inference/gpu/2.2.2/transformers/4.41.1/py311/Dockerfile
@@ -39,6 +39,9 @@ ARG HF_INFERENCE_TOOLKIT_VERSION=0.4.1
 ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
 RUN pip install "${HF_INFERENCE_TOOLKIT_URL}#egg=huggingface-inference-toolkit[torch,diffusers,st]"
 
+# Install google-cloud-storage required to download artifacts from Google Cloud Storage buckets
+RUN pip install --upgrade google-cloud-storage
+
 # copy entrypoint and change permissions
 COPY --chmod=0755 containers/pytorch/inference/gpu/2.2.2/transformers/4.41.1/py311/entrypoint.sh entrypoint.sh
 ENTRYPOINT ["bash", "-c", "./entrypoint.sh"]