]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
convert : register UMT5Model architecture for T5 conversion (#17160)
authorlevkropp <redacted>
Tue, 11 Nov 2025 08:38:30 +0000 (03:38 -0500)
committerGitHub <redacted>
Tue, 11 Nov 2025 08:38:30 +0000 (09:38 +0100)
Register UMT5Model as a supported architecture variant for T5 model conversion.
This allows the conversion to work for models downloaded with AutoModel.

convert_hf_to_gguf.py

index 13448fd68116cc4ec37601b16e709f755e1afa25..cc77a3db273e4b3901723bccf31bdff8b65d0a55 100755 (executable)
@@ -7354,6 +7354,7 @@ class PLMModel(TextModel):
 @ModelBase.register("T5ForConditionalGeneration")
 @ModelBase.register("MT5ForConditionalGeneration")
 @ModelBase.register("UMT5ForConditionalGeneration")
+@ModelBase.register("UMT5Model")
 class T5Model(TextModel):
     model_arch = gguf.MODEL_ARCH.T5