Use return_dict in RagModel forward pass
There were changes in the output format of models bu it looks like the RagModel forward pass was not updated to use `return_dict` as noticed in #8653 

I'm running the slow tests right now. If they pass I will update from draft pull request to open request
Closing since it was fixed in #8585