Make model recognized as `text-generation`? #1

No description provided.
julien-c changed pull request status to merged