Rag model huggingface. FloatTensor of shape (batch_size, sequence_length, config.