search

dmmiller612 / bert-extractive-summarizer

Easy to use extractive text summarization with BERT
MIT License
1.38k stars 305 forks source link

Using custom model: T5 and Bart #73

Open paulowoicho opened 4 years ago

paulowoicho commented 4 years ago

I tried using the tool with with a pretrained t5 model, but got the following error:

You have to specify either decoder_input_ids or decoder_inputs_embeds

I was wondering if the tool supports either of these models

dmmiller612 commented 4 years ago

Let me take a look. That would be a good idea to have both of those.

raytxz commented 3 years ago

Hi, any updates on this? Would appreciate to be able to use T5 with bert-extractive-summarizer.