Prompt Extend

Prompt Extend est un modèle de génération de texte basé sur GPT-2, entraîné sur un dataset de prompts stables de diffusion (stable diffusion prompts). Il génère des indications stylistiques adaptées à une idée principale de prompt, permettant d'améliorer la qualité et la cohérence des prompts utilisés en génération d'images ou de contenus créatifs. L'entraînement comprend des techniques avancées comme l'optimiseur Adam, un taux d'apprentissage adaptatif et un entraînement en précision mixte, assurant un modèle performant avec une bonne capacité de généralisation.