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

add grok to cloud models

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-09-07 09:20:58 +02:00
parent 0a94b2d8ae
commit fe03d7b38e
9 changed files with 210 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/caption/models_def.py
+10 -1
View File
@@ -81,6 +81,10 @@ vlm_models = {
f"Google Gemini 3.5 Flash Lite {ui_symbols.cloud}": "gemini-3.5-flash-lite",
f"Google Gemini 3.1 Flash Lite {ui_symbols.cloud}": "gemini-3.1-flash-lite",
f"Google Gemini 3.1 Pro {ui_symbols.cloud}": "gemini-3.1-pro-preview",
f"X.AI Grok 3 {ui_symbols.cloud}": "grok-3-latest",
f"X.AI Grok 3 Fast {ui_symbols.cloud}": "grok-3-fast-latest",
f"X.AI Grok 3 Mini {ui_symbols.cloud}": "grok-3-mini-latest",
f"X.AI Grok 3 Mini Fast {ui_symbols.cloud}": "grok-3-mini-fast-latest",
}
# Default model
@@ -224,5 +228,10 @@ Summary:
def get_vlm_repo(display_name: str) -> str:
"""Look up repo ID from display name, stripping any trailing symbols."""
from modules.logger import log
name = display_name.strip()
return vlm_models.get(name, name)
model = vlm_models.get(name, None)
if model is None:
log.warning(f"Model '{name}' not found")
return name
return model