The models by themselves return that when you pass return_dict=True (which will soon become the default). We can definitely add some code to carry on that type of outputs when the option is selected to the predictions.