diff --git a/keytotext/pipeline.py b/keytotext/pipeline.py index 4630517..9d2e3d9 100644 --- a/keytotext/pipeline.py +++ b/keytotext/pipeline.py @@ -7,7 +7,7 @@ PreTrainedModel, PreTrainedTokenizer, ) - +from .newmodels import NMPipeline class K2TPipeline: def __init__( @@ -87,6 +87,12 @@ def _tokenize( "model": "gagan3012/k2t-base", }, }, + "mrm8488/t5-base-finetuned-common_gen":{ + "impl": NMPipeline, + "default": { + "model": "mrm8488/t5-base-finetuned-common_gen", + }, + } }