hi @zanderbush, sure BART should also work for paraphrasing. Just fine-tune it on a paraphrasing dataset.
There’s a small mistake in the way you are using .generate
. If you want to do sampling you’ll need to set num_beams
to 0 and and do_sample
to True
. And set do_sample
to false and num_beams
to >1 for beam search. This post explains how to use generate