ソースを参照

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.
levkropp 2 ヶ月 前
コミット
2fc392ce35
1 ファイル変更1 行追加0 行削除
  1. 1 0
      convert_hf_to_gguf.py

+ 1 - 0
convert_hf_to_gguf.py

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