Text generation on GPU: Moved the encoded_prompt to correct device
Moved the `encoded_prompt` to the correct device to solve the problem when using GPU. 

Solves the problem mentioned in #227 #1414 #2360 
Thank you @alberduris !