Image Model Customization
LoRA Fine-Tuning
Efficiently create specialized models using Low-Rank Adaptation (LoRA). Fine-tune any compatible base model on the network with your own dataset to achieve unique styles and subjects.
Distributed Training Network
Leverage a global network of distributed GPUs for your training tasks. Our platform makes model customization accessible and scalable, without the need for dedicated high-end hardware.
Full Model Ownership
You retain complete ownership and control over your trained LoRA models. Download and deploy them anywhere, without restrictions.
{
"model": {
"name": "runwayml/stable-diffusion-v1-5",
"revision": "main"
},
"dataset": {
"name": "lambdalabs/naruto-blip-captions",
"image_column": "image",
"caption_column": "text"
},
"validation": {
"num_images": 4
},
"train_args": {
"learning_rate": "1e-4",
"batch_size": 1,
"gradient_accumulation_steps": 4,
"num_train_steps": 100,
"max_train_steps": 15000
},
"lora": {
"rank": 4,
"init_lora_weights": "gaussian",
"target_modules": ["to_k", "to_q", "to_v", "to_out.0"]
},
"transforms": {
"center_crop": true,
"random_flip": true
},
"dataloader_num_workers": 2,
"mixed_precision": "fp16",
"seed": 1337,
"checkpoint": null,
"version": "2.1.0"
}