diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 13448fd681..cc77a3db27 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -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