dreamshot-org/flux-dev-lora-trainer – Run with an API on Replicate

dreamshot-org / flux-dev-lora-trainer

Cold

Public
156 runs
A100 (80GB)

Iterate in playground

Run with an API

Playground API Examples Train README Versions

Input

prompt

*string

Shift + Return to add a new line

Prompt for generated image. If you include the `trigger_word` used in the training process you are more likely to activate the trained object, style, or concept in the resulting image.

image

file

Input image for img2img or inpainting mode. If provided, aspect_ratio, width, and height inputs are ignored.

mask

file

Input mask for inpainting mode. Black areas will be preserved, white areas will be inpainted. Must be provided along with 'image' for inpainting mode.

aspect_ratio

string

Aspect ratio for the generated image in text-to-image mode. The size will always be 1 megapixel, i.e. 1024x1024 if aspect ratio is 1:1. To use arbitrary width and height, set aspect ratio to 'custom'. Note: Ignored in img2img and inpainting modes.

Default: "1:1"

width

integer

(minimum: 256, maximum: 1440)

Width of the generated image in text-to-image mode. Only used when aspect_ratio=custom. Must be a multiple of 16 (if it's not, it will be rounded to nearest multiple of 16). Note: Ignored in img2img and inpainting modes.

height

integer

(minimum: 256, maximum: 1440)

Height of the generated image in text-to-image mode. Only used when aspect_ratio=custom. Must be a multiple of 16 (if it's not, it will be rounded to nearest multiple of 16). Note: Ignored in img2img and inpainting modes.

num_outputs

integer

(minimum: 1, maximum: 4)

Number of images to output.

Default: 1

lora_scale

number

(minimum: -1, maximum: 2)

Determines how strongly the main LoRA should be applied. Sane results between 0 and 1.

Default: 1

num_inference_steps

integer

(minimum: 1, maximum: 50)

Number of inference steps. More steps can give more detailed images, but take longer.

Default: 28

model

string

Which model to run inferences with. The dev model needs around 28 steps but the schnell model only needs around 4 steps.

Default: "dev"

guidance_scale

number

(minimum: 0, maximum: 10)

Guidance scale for the diffusion process. Lower values can give more realistic images. Good values to try are 2, 2.5, 3 and 3.5

Default: 3.5

prompt_strength

number

(minimum: 0, maximum: 1)

Prompt strength when using img2img / inpaint. 1.0 corresponds to full destruction of information in image

Default: 0.8

seed

integer

Random seed. Set for reproducible generation.

extra_lora

string

Shift + Return to add a new line

Combine this fine-tune with another LoRA. Supports Replicate models in the format <owner>/<username> or <owner>/<username>/<version>, HuggingFace URLs in the format huggingface.co/<owner>/<model-name>, CivitAI URLs in the format civitai.com/models/<id>[/<model-name>], or arbitrary .safetensors URLs from the Internet. For example, 'fofr/flux-pixar-cars'

extra_lora_scale

number

(minimum: -1, maximum: 2)

Determines how strongly the extra LoRA should be applied.

Default: 1

output_format

string

Format of the output images.

Default: "webp"

output_quality

integer

(minimum: 0, maximum: 100)

Quality when saving the output images, from 0 to 100. 100 is best quality, 0 is lowest quality. Not relevant for .png outputs

Default: 90

disable_safety_checker

boolean

Disable safety checker for generated images.

Default: false

Run this model in Node.js with one line of code:

npx create-replicate --model=dreamshot-org/flux-dev-lora-trainer

or set up a project from scratch

Install Replicate’s Node.js client library:

npm install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import and set up the client:

import Replicate from "replicate";
import fs from "node:fs";

const replicate = new Replicate({
  auth: process.env.REPLICATE_API_TOKEN,
});

Run dreamshot-org/flux-dev-lora-trainer using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "dreamshot-org/flux-dev-lora-trainer:fc37469fbd6d5d9787d2e7214251246189aa5e35205c29b2fb9350a4a5867fda",
  {
    input: {
      model: "dev",
      lora_scale: 1,
      num_outputs: 1,
      aspect_ratio: "1:1",
      output_format: "webp",
      guidance_scale: 3.5,
      output_quality: 90,
      prompt_strength: 0.8,
      extra_lora_scale: 1,
      num_inference_steps: 28
    }
  }
);

// To access the file URL:
console.log(output[0].url()); //=> "http://example.com"

// To write the file to disk:
fs.writeFile("my-image.png", output[0]);

To learn more, take a look at the guide on getting started with Node.js.

Install Replicate’s Python client library:

pip install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import the client:

import replicate

Run dreamshot-org/flux-dev-lora-trainer using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "dreamshot-org/flux-dev-lora-trainer:fc37469fbd6d5d9787d2e7214251246189aa5e35205c29b2fb9350a4a5867fda",
    input={
        "model": "dev",
        "lora_scale": 1,
        "num_outputs": 1,
        "aspect_ratio": "1:1",
        "output_format": "webp",
        "guidance_scale": 3.5,
        "output_quality": 90,
        "prompt_strength": 0.8,
        "extra_lora_scale": 1,
        "num_inference_steps": 28
    }
)
print(output)

To learn more, take a look at the guide on getting started with Python.

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Run dreamshot-org/flux-dev-lora-trainer using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

curl -s -X POST \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -H "Prefer: wait" \
  -d $'{
    "version": "dreamshot-org/flux-dev-lora-trainer:fc37469fbd6d5d9787d2e7214251246189aa5e35205c29b2fb9350a4a5867fda",
    "input": {
      "model": "dev",
      "lora_scale": 1,
      "num_outputs": 1,
      "aspect_ratio": "1:1",
      "output_format": "webp",
      "guidance_scale": 3.5,
      "output_quality": 90,
      "prompt_strength": 0.8,
      "extra_lora_scale": 1,
      "num_inference_steps": 28
    }
  }' \
  https://api.replicate.com/v1/predictions

To learn more, take a look at Replicate’s HTTP API reference docs.

Output

No output yet! Press "Submit" to start a prediction.

Run time and cost

This model runs on Nvidia A100 (80GB) GPU hardware. We don't yet have enough runs of this model to provide performance information.

Readme

This model doesn't have a readme.