File size: 221 Bytes
a9eff47
fa0655e
a9eff47
fa0655e
a9eff47
fa0655e
a9eff47
1
2
3
4
5
6
7
MODEL_NAME = 'nonamenlp/news_gen' 

TOKENIZER_NAME = "nonamenlp/news_gen"

trained_model = MT5ForConditionalGeneration.from_pretrained(MODEL_NAME, return_dict=True)

tokenizer = T5Tokenizer.from_pretrained(TOKENIZER_NAME)