*optional*):
            Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
a