add T5GemmaEncoder to init

alien-0119 · alien-0119 · commit 4e9aae79800d · 2025-11-10T15:50:46.000+08:00
diff --git a/mindone/transformers/__init__.py b/mindone/transformers/__init__.py
@@ -1319,6 +1319,7 @@
     T5PreTrainedModel,
 )
 from .models.t5gemma import (
+    T5GemmaEncoder,
     T5GemmaEncoderModel,
     T5GemmaForConditionalGeneration,
     T5GemmaForSequenceClassification,
diff --git a/mindone/transformers/models/t5gemma/modeling_t5gemma.py b/mindone/transformers/models/t5gemma/modeling_t5gemma.py
@@ -1366,6 +1366,7 @@ def construct(
 __all__ = [
     "T5GemmaForConditionalGeneration",
     "T5GemmaModel",
+    "T5GemmaEncoder", # for diffusers
     "T5GemmaEncoderModel",
     "T5GemmaPreTrainedModel",
     "T5GemmaForSequenceClassification",

Original file line number	Diff line number	Diff line change
`@@ -1319,6 +1319,7 @@`
`1319`	`1319`	`T5PreTrainedModel,`
`1320`	`1320`	`)`
`1321`	`1321`	`from .models.t5gemma import (`
	`1322`	`+ T5GemmaEncoder,`
`1322`	`1323`	`T5GemmaEncoderModel,`
`1323`	`1324`	`T5GemmaForConditionalGeneration,`
`1324`	`1325`	`T5GemmaForSequenceClassification,`