train_build_image
Build the headless GPU trainer Docker image from Dockerfile, using a specified ai-toolkit commit or tag for reproducibility. Requires a running Docker daemon.
Instructions
Build the headless GPU trainer image (comfyui-mcp-trainer:latest) from docker/trainer/Dockerfile — one-time, several minutes (CUDA + torch + ai-toolkit). Requires a reachable docker daemon. aiToolkitRef pins the ai-toolkit commit/tag for reproducibility.
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| aiToolkitRef | No | ai-toolkit git ref (commit/tag) to build against. Default: the Dockerfile's pinned ref. |