From 8d8d81e6cd2646bba76a87b63b78fbff7ee128d1 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Mon, 30 Mar 2026 14:31:28 +0800
Subject: [PATCH] Update t5_gemma_model.py

---
 inference/model/t5_gemma/t5_gemma_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference/model/t5_gemma/t5_gemma_model.py b/inference/model/t5_gemma/t5_gemma_model.py
index da4ad0c..f98384f 100644
--- a/inference/model/t5_gemma/t5_gemma_model.py
+++ b/inference/model/t5_gemma/t5_gemma_model.py
@@ -7,7 +7,7 @@ from transformers import AutoTokenizer
 try :
     from transformers.models.t5gemma.modeling_t5gemma import T5GemmaEncoderModel,T5GemmaConfig
 except:
-    from transformers.models.t5gemma import T5GemmaEncoderModel
+    from transformers.models.t5gemma import T5GemmaEncoderModel,T5GemmaConfig
 from transformers import AutoModel, AutoTokenizer,AutoConfig
 from safetensors.torch import load_file
 from ...common import CPUOffloadWrapper, get_arch_memory