Hi Patrick @patrickvonplaten,
Thanks for your great work!! I have a question.
I now almost finish translating TFDPR 1 , and currently also working on TFRag (I already made TFRagModel works like Pytorch, and I am investigating TFRagTokenForGeneration right now) .
Since the current code of RagTokenForGeneration involves with both _no_beam_search_generation  and  _beam_search_generation , regarding the Refactor, I think I should just go ahead with the current version, and fix them later for refactor.
Or would you suggest that I wait for Pytorch Refactor, or wait for both Pytorch & TF Refactor ?