Siddharth63
/

pubmedul2-tiny-nl6

text2text-generation

encoder-decoder

text-generation-inference

Model card Files Files and versions

Siddharth63 commited on Sep 19, 2023

Commit

d0287d4

·

1 Parent(s): a220878

Update ul2_tasks.py

Files changed (1) hide show

ul2_tasks.py +1 -1

ul2_tasks.py CHANGED Viewed

@@ -26,7 +26,7 @@ S_DENOISER_TOKEN_PREFIX = "[S2S]"
 TaskRegistry = seqio.TaskRegistry
-vocabulary = seqio.SentencePieceVocabulary('spiece.model', extra_ids=128)
 DEFAULT_OUTPUT_FEATURES = {
     "inputs": seqio.Feature(vocabulary=vocabulary, add_eos=True, required=False),

 TaskRegistry = seqio.TaskRegistry
+vocabulary = seqio.SentencePieceVocabulary('spiece.model')
 DEFAULT_OUTPUT_FEATURES = {
     "inputs": seqio.Feature(vocabulary=vocabulary, add_eos=True, required=False),