Skip to content

Commit

Permalink
Update pipeline.py
Browse files Browse the repository at this point in the history
  • Loading branch information
gagan3012 committed May 10, 2021
1 parent 46827d8 commit 43078ef
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion keytotext/pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
PreTrainedModel,
PreTrainedTokenizer,
)

from .newmodels import NMPipeline

class K2TPipeline:
def __init__(
Expand Down Expand Up @@ -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",
},
}
}


Expand Down

0 comments on commit 43078ef

Please sign in to comment.