submit_training_job
Destructive
Queue an AI Toolkit training job (typically LoRA) by supplying a YAML config and GPU type/count. Returns once the job is queued; use status/result tools to monitor and retrieve outputs.
Instructions
Submit an AI Toolkit training job (typically LoRA training).
Backs ``POST /prod/v1/trainers/ai-toolkit/jobs``. The job mounts a
``READY`` dataset and runs the config you supply. Training runs for
hours — this returns as soon as the job is queued; track it with
``get_training_job_status`` and pull artifacts with
``get_training_job_result``.
Args:
config_file: The complete AI Toolkit YAML config as a string.
Two paths in it are fixed by the platform:
``training_folder`` must be ``/app/ai-toolkit/output``, and
the dataset's ``folder_path`` must be
``/app/ai-toolkit/datasets/{dataset_name}`` where
``dataset_name`` is the dataset's ``name`` (not its id).
gpu_type: ``ADA_80_PLUS`` (H100) or ``HOPPER_141`` (H200).
gpu_count: 1 for single-GPU (default), or 8 for multi-GPU.
Multi-GPU is only supported on ``ADA_80_PLUS``.
gpu_id: Optional specific GPU selector, e.g. ``"#1"``.
Input Schema
TableJSON Schema
| Name | Required | Description | Default |
|---|---|---|---|
| gpu_id | No | ||
| gpu_type | No | ADA_80_PLUS | |
| gpu_count | No | ||
| config_file | Yes |