From 666b75ea8718efdf67909c5a2d5b4a96bce6309c Mon Sep 17 00:00:00 2001
From: dstnluong-google <129889805+dstnluong-google@users.noreply.github.com>
Date: Wed, 21 Feb 2024 21:58:46 +0000
Subject: [PATCH] Move GCS install to requirements files.

---
 Dockerfile                     | 3 ---
 server/requirements_common.txt | 1 +
 server/requirements_cuda.txt   | 1 +
 server/requirements_rocm.txt   | 1 +
 4 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index b6560d39..e79372a3 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -218,9 +218,6 @@ COPY --from=mamba-builder /usr/src/causal-conv1d/build/lib.linux-x86_64-cpython-
 # Install flash-attention dependencies
 RUN pip install einops --no-cache-dir
 
-# Install GCS library
-RUN pip install --upgrade google-cloud-storage
-
 # Install server
 COPY proto proto
 COPY server server
diff --git a/server/requirements_common.txt b/server/requirements_common.txt
index 5a321834..696c1150 100644
--- a/server/requirements_common.txt
+++ b/server/requirements_common.txt
@@ -8,6 +8,7 @@ einops==0.6.1 ; python_version >= "3.9" and python_version < "3.13"
 filelock==3.13.1 ; python_version >= "3.9" and python_version < "3.13"
 fsspec==2023.10.0 ; python_version >= "3.9" and python_version < "3.13"
 googleapis-common-protos==1.61.0 ; python_version >= "3.9" and python_version < "3.13"
+google-cloud-storage==2.14.0 ; python_version >= "3.9" and python_version < "3.13"
 grpc-interceptor==0.15.4 ; python_version >= "3.9" and python_version < "3.13"
 grpcio-reflection==1.59.3 ; python_version >= "3.9" and python_version < "3.13"
 grpcio-status==1.59.3 ; python_version >= "3.9" and python_version < "3.13"
diff --git a/server/requirements_cuda.txt b/server/requirements_cuda.txt
index 1e3477bf..d9e75abf 100644
--- a/server/requirements_cuda.txt
+++ b/server/requirements_cuda.txt
@@ -9,6 +9,7 @@ einops==0.6.1 ; python_version >= "3.9" and python_version < "3.13"
 filelock==3.13.1 ; python_version >= "3.9" and python_version < "3.13"
 fsspec==2023.10.0 ; python_version >= "3.9" and python_version < "3.13"
 googleapis-common-protos==1.62.0 ; python_version >= "3.9" and python_version < "3.13"
+google-cloud-storage==2.14.0 ; python_version >= "3.9" and python_version < "3.13"
 grpc-interceptor==0.15.4 ; python_version >= "3.9" and python_version < "3.13"
 grpcio-reflection==1.60.1 ; python_version >= "3.9" and python_version < "3.13"
 grpcio-status==1.60.1 ; python_version >= "3.9" and python_version < "3.13"
diff --git a/server/requirements_rocm.txt b/server/requirements_rocm.txt
index 3912abd8..718dd972 100644
--- a/server/requirements_rocm.txt
+++ b/server/requirements_rocm.txt
@@ -8,6 +8,7 @@ einops==0.6.1 ; python_version >= "3.9" and python_version < "3.13"
 filelock==3.13.1 ; python_version >= "3.9" and python_version < "3.13"
 fsspec==2023.10.0 ; python_version >= "3.9" and python_version < "3.13"
 googleapis-common-protos==1.62.0 ; python_version >= "3.9" and python_version < "3.13"
+google-cloud-storage==2.14.0 ; python_version >= "3.9" and python_version < "3.13"
 grpc-interceptor==0.15.4 ; python_version >= "3.9" and python_version < "3.13"
 grpcio-reflection==1.60.1 ; python_version >= "3.9" and python_version < "3.13"
 grpcio-status==1.60.1 ; python_version >= "3.9" and python_version < "3.13"