Train a Dreambooth Model with Custom Images

• fine_tune
• fine_tune_v2
• fine_tune_status
• cancle_training
• finetune_list
• finetune/delete
• base64_crop
• lora_finetune

Overview

Using this endpoint you can train a Dreambooth model with your own images. You can train a model on any object or person.

Request

--request POST 'https://stablediffusionapi.com/api/v3/fine_tune' \

Make a POST request to https://stablediffusionapi.com/api/v3/fine_tune endpoint and pass the required parameters as a request body.

Now you can train a model on any other models used as a base model and get its style. You can choose to train on models like midjourney, portraitplus and many more.

info

Get the base_model_id from here.

tip

You can also read our blog article to learn more about model training.

Body Attributes

Parameter	Description
key	Your API Key used for request authorization
instance_prompt	Text prompt with how you want to call your trained person/object
class_prompt	Classification of the trained person/object
base_model_id	Base model ID to train your model on
images	Pass accessible direct links to images, cropped to 512 x 512 px. A good number is about 7-8 images.
training_type	The type of the object you are training on
max_train_steps	Set it at 2 times the number of images (Ni*2; maximum value 2000)
webhook	Get a post call when training is complete

Training Types

The table below lists all the possible values for the training_type parameter.

Value	Description
men	Train on faces of men.
female	Train on faces of females.
couple	Train on couples of male and female; in images array pass images of couples, instead of images of a single person.
null	Train on object or anything.

Webhook Post JSON

This is an example webhook post call in JSON format.

{
 "status": "success",
 "training_status": "deploying_gpu",
 "logs": "it will take upto 25 minutes",
 "model_id": "F5jvdzGnYi",
}

Training Status Values

The table below describes all possible training statuses.

Status	Description
deploying_gpu	Deploying GPU
training_started	Training started
training_success	Training completed successfully
trained_model_compressing	Compressing the trained model
trained_model_uploading	Uploading the trained model
trained_model_uploaded	Trained model uploaded
deploying_model	Deploying the trained model
model_ready	The trained model is ready for use

Example

Body

Body

{
  "key": "",
  "instance_prompt": "photo of adhik person",
  "class_prompt": "photo of person",
  "base_model_id": "portraitplus-diffusion",
  "images": [
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00001.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00002.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00003.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00004.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00005.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00006.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00007.png"
  ],
  "seed": "0",
  "training_type": "men",
  "max_train_steps": "2000",
  "webhook": ""
}

Request

JS

var myHeaders = new Headers();
myHeaders.append("Content-Type", "application/json");

var raw = JSON.stringify({
  "key": "",
  "instance_prompt": "photo of adhik person",
  "class_prompt": "photo of person",
  "base_model_id": "portraitplus-diffusion",
  "images": [
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00001.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00002.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00003.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00004.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00005.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00006.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00007.png"
  ],
  "seed": "0",
  "training_type": "men",
  "max_train_steps": "2000",
  "webhook": ""
});

var requestOptions = {
  method: 'POST',
  headers: myHeaders,
  body: raw,
  redirect: 'follow'
};

fetch("https://stablediffusionapi.com/api/v3/fine_tune", requestOptions)
  .then(response => response.text())
  .then(result => console.log(result))
  .catch(error => console.log('error', error));

PHP

<?php

$payload = [
  "key" => "", 
  "instance_prompt" => "photo of adhik person", 
  "class_prompt" => "photo of person", 
  "base_model_id" => "portraitplus-diffusion", 
  "images" => [
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00001.png", 
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00002.png", 
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00003.png", 
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00004.png", 
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00005.png", 
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00006.png", 
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00007.png" 
  ], 
  "seed" => "0", 
  "training_type" => "men", 
  "max_train_steps" => "2000", 
  "webhook" => "" 
];

$curl = curl_init();

curl_setopt_array($curl, array(
  CURLOPT_URL => 'https://stablediffusionapi.com/api/v3/fine_tune',
  CURLOPT_RETURNTRANSFER => true,
  CURLOPT_ENCODING => '',
  CURLOPT_MAXREDIRS => 10,
  CURLOPT_TIMEOUT => 0,
  CURLOPT_FOLLOWLOCATION => true,
  CURLOPT_HTTP_VERSION => CURL_HTTP_VERSION_1_1,
  CURLOPT_CUSTOMREQUEST => 'POST',
  CURLOPT_POSTFIELDS => json_encode($payload),
  CURLOPT_HTTPHEADER => array(
    'Content-Type: application/json'
  ),
));

$response = curl_exec($curl);

curl_close($curl);
echo $response;

NODE

var request = require('request');
var options = {
  'method': 'POST',
  'url': 'https://stablediffusionapi.com/api/v3/fine_tune',
  'headers': {
    'Content-Type': 'application/json'
  },
  body: JSON.stringify({
    "key": "",
    "instance_prompt": "photo of adhik person",
    "class_prompt": "photo of person",
    "base_model_id": "portraitplus-diffusion",
    "images": [
      "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00001.png",
      "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00002.png",
      "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00003.png",
      "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00004.png",
      "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00005.png",
      "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00006.png",
      "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00007.png"
    ],
    "seed": "0",
    "training_type": "men",
    "max_train_steps": "2000",
    "webhook": ""
  })
};

request(options, function (error, response) {
  if (error) throw new Error(error);
  console.log(response.body);
});

PYTHON

import requests
import json

url = "https://stablediffusionapi.com/api/v3/fine_tune"

payload = json.dumps({
  "key": "",
  "instance_prompt": "photo of adhik person",
  "class_prompt": "photo of person",
  "base_model_id": "portraitplus-diffusion",
  "images": [
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00001.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00002.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00003.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00004.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00005.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00006.png",
    "https://stablediffusionapi.com/storage/avatars/AdhikJoshi_00007.png"
  ],
  "seed": "0",
  "training_type": "men",
  "max_train_steps": "2000",
  "webhook": ""
})

headers = {
  'Content-Type': 'application/json'
}

response = requests.request("POST", url, headers=headers, data=payload)

print(response.text)

JAVA

OkHttpClient client = new OkHttpClient().newBuilder()
  .build();
MediaType mediaType = MediaType.parse("application/json");
RequestBody body = RequestBody.create(mediaType, "{\n    \"key\": \"\",\n    \"instance_prompt\": \"a photo of adhik person\",\n    \"class_prompt\" : \"a photo of person\",\n    \"base_model_id\" : \"portraitplus-diffusion\",\n    \"images\": [\n        \"https://stablediffusionapi.com/storage/generations/AdhikJoshi_00000.png\",\n        \"https://stablediffusionapi.com/storage/generations/AdhikJoshi_00001.png\",\n        \"https://stablediffusionapi.com/storage/generations/AdhikJoshi_00002.png\",\n        \"https://stablediffusionapi.com/storage/generations/AdhikJoshi_00003.png\",\n        \"https://stablediffusionapi.com/storage/generations/AdhikJoshi_00004.png\",\n        \"https://stablediffusionapi.com/storage/generations/AdhikJoshi_00005.png\",\n        \"https://stablediffusionapi.com/storage/generations/AdhikJoshi_00006.png\",\n        \"https://stablediffusionapi.com/storage/generations/AdhikJoshi_00007.png\"\n    ],\n    \"seed\": \"0\",\n   \"training_type\": \"men\",\n  \"max_train_steps\": \"2000\",\n  \"webhook\": \"\"\n}");

Request request = new Request.Builder()
  .url("https://stablediffusionapi.com/api/v3/fine_tune")
  .method("POST", body)
  .addHeader("Content-Type", "application/json")
  .build();
Response response = client.newCall(request).execute();

Response

{
  "status": "success",
  "messege": "deploying_gpu",
  "data": "it will take upto 25 minutes.",
  "training_id": "F5jvdzGnYi"
}