Skip to content

Latest commit

 

History

History
84 lines (61 loc) · 4.98 KB

File metadata and controls

84 lines (61 loc) · 4.98 KB

Models

Models can be downloaded using the built-in downloads manager on the Captions tab (📥💾 button).

Screenshot Model Manager

  • To download a model simply click the "Download" button, the model will be downloaded into the /models folder.
  • Note that for "Gated" models like SAM3 you will need to log in to HuggingFace first and generate an Authentication Token. Check the info at the top of the window for more details.

Supported Model Families

VisionCaptioner currently supports two Vision-Language Model families:

  • Qwen-VL (Qwen2.5-VL, Qwen3-VL) — developed by the Qwen Team at Alibaba Cloud.
  • Google Gemma 4 (E2B, E4B, 26B-A4B MoE, 31B) — developed by Google DeepMind.

Abliterated (uncensored) variants of both families should work as well, since they share the same architecture as the base models.

Manual Download

Alternatively, you can manually download models from HuggingFace into the /models folder.

Qwen-VL Models

Model Link
Qwen2.5-VL-3B-Instruct HuggingFace link
Qwen2.5-VL-7B-Instruct HuggingFace link
Qwen2.5-VL Abliterated HuggingFace link
Qwen2.5-VL Abliterated Caption-It HuggingFace link
Qwen3-VL-2B-Instruct HuggingFace link
Qwen3-VL-4B-Instruct HuggingFace link
Qwen3-VL-8B-Instruct HuggingFace link
Qwen3-VL-32B-Instruct HuggingFace link
Qwen3-VL Abliterated HuggingFace link

Google Gemma 4 Models

Model Link
Gemma-4-E2B-it HuggingFace link
Gemma-4-E4B-it HuggingFace link
Gemma-4-26B-A4B-it HuggingFace link
Gemma-4-31B-it HuggingFace link
Gemma 4 Abliterated HuggingFace link

Gemma 4 Specific Settings

When a Gemma 4 model is selected, an extra Vision Tokens dropdown becomes active in the Captions tab. This controls the soft visual token budget per image:

Budget Use case
70 Fastest, lowest VRAM, coarse detail
140 Fast
280 Default — good balance
560 Detailed
1120 Maximum detail, highest VRAM

The "Max Resolution" setting is ignored for Gemma 4 — Gemma's processor handles its own resizing based on the vision token budget. For Qwen models the Vision Tokens dropdown is greyed out and "Max Resolution" controls detail instead.

Note: Gemma 4's built-in "thinking" mode is automatically disabled for captioning, since reasoning tokens add latency without improving caption quality.

GGUF Models

Models in GGUF format are supported for the Qwen-VL and Gemma 4 families. GGUF support requires the llama-cpp-python package from JamePeng/llama-cpp-python.

Automatic Install (recommended)

When you try to load a GGUF model without llama-cpp-python installed, VisionCaptioner will offer to install it for you. It automatically detects your Python version, operating system, and CUDA version, then downloads and installs the matching wheel from GitHub. The log shows exactly which package was selected so you can verify the choice.

Manual Install

If you prefer to install manually:

  • pip install llama-cpp-python does not work!
  • You need the latest version from JamePeng/llama-cpp-python on GitHub
  • Pick the wheel that matches your system:
    • CUDA version: cu124, cu126, cu128, or cu130 (check via python -c "import torch; print(torch.version.cuda)")
    • Platform: win (Windows), linux, or Metal (macOS)
    • Python version: cp310, cp311, cp312, cp313, etc.
  • Install with: pip install <url-to-wheel-file>

GGUF Model Notes

  • Make sure you download the GGUF version of the model and don't forget the accompanying mmproj file.

Gemma 4 GGUF

Gemma 4 GGUF models (e.g. from unsloth/gemma-4-*-GGUF) are supported and require llama-cpp-python v0.3.35 or newer from JamePeng/llama-cpp-python. Older versions do not include the Gemma4ChatHandler and will show an error asking you to update. The Vision Token Budget setting does not apply to GGUF models — it is automatically greyed out. Thinking mode is handled internally by the chat handler.