21 lines
778 B
Python
21 lines
778 B
Python
from __future__ import annotations
|
|
|
|
MAGIC_PROMPT_MODELS = [
|
|
"Gustavosta/MagicPrompt-Stable-Diffusion",
|
|
"daspartho/prompt-extend",
|
|
"succinctly/text2image-prompt-generator",
|
|
"microsoft/Promptist",
|
|
"AUTOMATIC/promptgen-lexart",
|
|
"AUTOMATIC/promptgen-majinai-safe",
|
|
"AUTOMATIC/promptgen-majinai-unsafe",
|
|
"kmewhort/stable-diffusion-prompt-bolster",
|
|
"Gustavosta/MagicPrompt-Dalle",
|
|
"Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator",
|
|
"Ar4ikov/gpt2-medium-650k-stable-diffusion-prompt-generator",
|
|
"crumb/bloom-560m-RLHF-SD2-prompter-aesthetic",
|
|
"Meli/GPT2-Prompt",
|
|
"DrishtiSharma/StableDiffusion-Prompt-Generator-GPT-Neo-125M",
|
|
]
|
|
DEFAULT_MAGIC_MODEL = MAGIC_PROMPT_MODELS[0]
|
|
OPTION_WRITE_RAW_TEMPLATE = "pp_write_raw_template"
|