Skip to content

Commit d5d00bb

Browse files
committed
Add t5gemma2 to image-text-to-text category.
1 parent bd4c480 commit d5d00bb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/transformers/models/auto/modeling_auto.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1062,6 +1062,7 @@ class _BaseModelWithGenerate(PreTrainedModel, GenerationMixin):
10621062
("qwen3_vl_moe", "Qwen3VLMoeForConditionalGeneration"),
10631063
("shieldgemma2", "Gemma3ForConditionalGeneration"),
10641064
("smolvlm", "SmolVLMForConditionalGeneration"),
1065+
("t5gemma2", "T5Gemma2ForConditionalGeneration"),
10651066
("udop", "UdopForConditionalGeneration"),
10661067
("video_llama_3", "VideoLlama3ForConditionalGeneration"),
10671068
("vipllava", "VipLlavaForConditionalGeneration"),

0 commit comments

Comments
 (0)