AI Trainer - RunPod Serverless
Multi-model LoRA training service using ai-toolkit.
Supported Models
| Model Key | Description | Base Model |
|---|---|---|
| wan21_1b | Wan2.1 1.3B Video | Wan-AI/Wan2.1-T2V-1.3B-Diffusers |
| wan21_14b | Wan2.1 14B Video | Wan-AI/Wan2.1-T2V-14B-Diffusers |
| wan22_14b | Wan2.2 14B Video | ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16 |
| qwen_image | Qwen Image Gen | Qwen/Qwen-Image |
| qwen_image_edit | Qwen Image Edit | Qwen/Qwen-Image-Edit |
| flux_dev | FLUX.1 Dev | black-forest-labs/FLUX.1-dev |
| flux_schnell | FLUX.1 Schnell | black-forest-labs/FLUX.1-schnell |
API Usage
List Models
Check Status
Train LoRA
RunPod Deployment
Environment Variables
- HF_TOKEN: HuggingFace token for gated models
Model Caching
For faster cold starts, set one of these HuggingFace repos in the RunPod "Model" field:
- FLUX: black-forest-labs/FLUX.1-dev
- Wan 2.2: ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16
- Qwen: Qwen/Qwen-Image
Note: RunPod only supports ONE cached model per endpoint.