auto_caption_dataset
Queues the built-in captioning job for a LoRA training dataset, captioning only uncaptioned files by default and supporting full recaptioning or a custom prompt.
Instructions
Queue ai-toolkit's built-in captioning job (Qwen3-Omni) for a dataset, like the UI's Caption button.
Only uncaptioned files are captioned unless recaption is set. prompt overrides the captioning
instruction; by default an image prompt is used for image-only datasets and ai-toolkit's video/audio
prompt otherwise. Uses GPU time and downloads the captioner model on first use. Track it with get_job.
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| prompt | No | ||
| dataset | Yes | ||
| gpu_ids | No | 0 | |
| recaption | No | ||
| caption_ext | No | txt |
Output Schema
| Name | Required | Description | Default |
|---|---|---|---|
| result | Yes |