Multi-decoder text generation with BART
Hello, I would like to perform text generation with BART, which takes a single text input, and uses 2 decoders to generate two different outputs from the single input, in a multi-decoder setup. I would like to apply a constraint such asprefix_allowed_tokens_fnonly for one of the two decoders, and use unconstrained generation for the other decoder. So my question is: is it possible to have two calls to themodel.generate()function in such a multi-decoder setup, one for each decoder? One call (for the first decoder) tomodel.generate()would take theprefix_allowed_tokens_fnas a parameter input, and the other call (for the second decoder) would not.Thank you very much!