|
|
@@ -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
|
|
|
|