// the find
AUTOMATIC1111/stable-diffusion-webui-promptgen
stable-diffusion-webui-promptgen
This is a small extension for AUTOMATIC1111's webui that generates Stable Diffusion prompts using distilgpt2 models finetuned on scraped prompt datasets (lexica.art, majinai.art). It's for webui users who want prompt suggestions or inspiration without leaving the UI.
Does one narrow thing and does it simply — no bloat, no extra dependencies beyond what webui already has. Supports both pulling models from huggingface at runtime and fully offline use by dropping model files into a local directory, which matters for people running airgapped or on slow connections. The base models (distilgpt2) are tiny, so generation is fast even on CPU.
Last commit is from July 2023 — webui's extension API and UI internals have moved on since then, so there's a real chance this breaks on current webui without someone patching it. The included models are finetuned on scraped datasets of unknown licensing and quality; one is explicitly named 'majinai-unsafe' with no filtering described, so output quality and content are entirely dependent on what got scraped. distilgpt2 is a weak base model for this task — expect generic, repetitive prompt suggestions rather than anything that meaningfully improves your results over hand-writing prompts. No tests, no CI, and the README is the only documentation.