From 03fb94b8538489eb7ce465aa2368e1198c7b8dde Mon Sep 17 00:00:00 2001
From: dstnluong-google <129889805+dstnluong-google@users.noreply.github.com>
Date: Wed, 21 Feb 2024 22:03:55 +0000
Subject: [PATCH] gs:// model_id is already set to /tmp/gcs_model/

---
 server/text_generation_server/server.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/server/text_generation_server/server.py b/server/text_generation_server/server.py
index c5cd40d8..d5adbd32 100644
--- a/server/text_generation_server/server.py
+++ b/server/text_generation_server/server.py
@@ -192,8 +192,6 @@ def serve(
             local_url = unix_socket_template.format(uds_path, 0)
             server_urls = [local_url]
 
-        if model_id.startswith("gs://"):
-            model_id = "/tmp/gcs_model"
         try:
             model = get_model(
                 model_id,