Fix text-generation pipeline and move chat_input to top level

134e302

verified

Phoenixak99 commited on Feb 11