Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

add gemma3

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-03-15 15:30:57 -04:00
parent a5d3a68107
commit dbfd59434f
14 changed files with 108 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/shared.py
+2 -1
View File
@@ -19,7 +19,7 @@ from modules.dml import memory_providers, default_memory_provider, directml_do_h
from modules.onnx_impl import initialize_onnx, execution_providers
from modules.memstats import memory_stats, ram_stats # pylint: disable=unused-import
from modules.interrogate.openclip import caption_models, caption_types, get_clip_models, refresh_clip_models, category_types
from modules.interrogate.vqa import vlm_models, vlm_prompts
from modules.interrogate.vqa import vlm_models, vlm_prompts, vlm_system
from modules.ui_components import DropdownEditable
from modules.options import OptionInfo
import modules.memmon
@@ -855,6 +855,7 @@ options_templates.update(options_section(('interrogate', "Interrogate"), {
"interrogate_vlm_sep": OptionInfo("<h2>VLM</h2>", "", gr.HTML),
"interrogate_vlm_model": OptionInfo(list(vlm_models)[0], "VLM: default model", gr.Dropdown, {"choices": list(vlm_models)}),
"interrogate_vlm_prompt": OptionInfo(vlm_prompts[2], "VLM: default prompt", DropdownEditable, {"choices": vlm_prompts }),
"interrogate_vlm_system": OptionInfo(vlm_system, "VLM: default prompt"),
"interrogate_vlm_num_beams": OptionInfo(3, "VLM: num beams", gr.Slider, {"minimum": 1, "maximum": 16, "step": 1, "visible": False}),
"interrogate_vlm_max_length": OptionInfo(512, "VLM: max length", gr.Slider, {"minimum": 1, "maximum": 4096, "step": 1, "visible": False}),
"interrogate_vlm_do_sample": OptionInfo(False, "VLM: use sample method"),