From c72d993ff4a3bdccfb0d6c439809c77eaa6e01d7 Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Sun, 26 Oct 2025 03:12:10 +0000 Subject: [PATCH] Prompt enhance update Added Qwen3-VL models to prompt enhance Added Qwen2.5-VL and Glitter Gemma 4B to img2img prompt enhance --- scripts/prompt_enhance.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/scripts/prompt_enhance.py b/scripts/prompt_enhance.py index 221d815dc..87396136f 100644 --- a/scripts/prompt_enhance.py +++ b/scripts/prompt_enhance.py @@ -32,6 +32,14 @@ def b64(image): class Options: img2img = [ 'google/gemma-3-4b-it', + 'allura-org/Gemma-3-Glitter-4B', + 'Qwen/Qwen2.5-VL-3B-Instruct', + 'Qwen/Qwen3-VL-2B-Instruct', + 'Qwen/Qwen3-VL-2B-Thinking', + 'Qwen/Qwen3-VL-4B-Instruct', + 'Qwen/Qwen3-VL-4B-Thinking', + 'Qwen/Qwen3-VL-8B-Instruct', + 'Qwen/Qwen3-VL-8B-Thinking', ] models = { 'google/gemma-3-1b-it': {}, @@ -49,6 +57,12 @@ class Options: 'Qwen/Qwen2.5-1.5B-Instruct': {}, 'Qwen/Qwen2.5-3B-Instruct': {}, 'Qwen/Qwen2.5-VL-3B-Instruct': {}, + 'Qwen/Qwen3-VL-2B-Instruct': {}, + 'Qwen/Qwen3-VL-2B-Thinking': {}, + 'Qwen/Qwen3-VL-4B-Instruct': {}, + 'Qwen/Qwen3-VL-4B-Thinking': {}, + 'Qwen/Qwen3-VL-8B-Instruct': {}, + 'Qwen/Qwen3-VL-8B-Thinking': {}, 'microsoft/Phi-4-mini-instruct': {}, 'HuggingFaceTB/SmolLM2-135M-Instruct': {}, 'HuggingFaceTB/SmolLM2-360M-Instruct': {},