Skip to content

Commit 2fc392c

Browse files
authored
convert : register UMT5Model architecture for T5 conversion (#17160)
Register UMT5Model as a supported architecture variant for T5 model conversion. This allows the conversion to work for models downloaded with AutoModel.
1 parent ece0f5c commit 2fc392c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

convert_hf_to_gguf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7354,6 +7354,7 @@ def prepare_tensors(self):
73547354
@ModelBase.register("T5ForConditionalGeneration")
73557355
@ModelBase.register("MT5ForConditionalGeneration")
73567356
@ModelBase.register("UMT5ForConditionalGeneration")
7357+
@ModelBase.register("UMT5Model")
73577358
class T5Model(TextModel):
73587359
model_arch = gguf.MODEL_ARCH.T5
73597360

0 commit comments

Comments
 (0)