From ee9b5a2be61d53999aab4b23777ad95e9d5cc102 Mon Sep 17 00:00:00 2001
From: dstnluong-google <129889805+dstnluong-google@users.noreply.github.com>
Date: Wed, 21 Feb 2024 21:54:31 +0000
Subject: [PATCH] nit: Rename to Gemma

---
 server/text_generation_server/models/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/text_generation_server/models/__init__.py b/server/text_generation_server/models/__init__.py
index abab3486..db58d6bf 100644
--- a/server/text_generation_server/models/__init__.py
+++ b/server/text_generation_server/models/__init__.py
@@ -327,7 +327,7 @@ def get_model(
             )
         elif sharded:
             raise NotImplementedError(
-                FLASH_ATT_ERROR_MESSAGE.format("Sharded Golden Gate")
+                FLASH_ATT_ERROR_MESSAGE.format("Sharded Gemma")
             )
         else:
             return CausalLM(