Add Qwen3.6 VL captioner.

This commit is contained in:
Jaret Burkett
2026-07-15 07:02:33 -06:00
parent 18da85153b
commit 691ddf434e
5 changed files with 68 additions and 13 deletions

View File

@@ -40,6 +40,7 @@ class CaptionConfig:
self.recaption = kwargs.get("recaption", False)
self.max_res = kwargs.get("max_res", 512)
self.max_new_tokens = kwargs.get("max_new_tokens", 128)
self.thinking = kwargs.get("thinking", False)
self.caption_prompt = kwargs.get(
"caption_prompt", "Describe this image in detail."
)