alexgenovese / autotrain-dreambooth

Auto Train Dreambooth SDXL | WIP - Not working

  • Public
  • 48 runs
  • L40S
  • GitHub
  • License
Iterate in playground
Run with an API
Playground API Examples README Versions

Input

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

npx create-replicate --model=alexgenovese/autotrain-dreambooth
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";

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

Run alexgenovese/autotrain-dreambooth using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "alexgenovese/autotrain-dreambooth:e3e8ff8604f3101092341004f3cd9ebaf3fc92b9b1fc46725ac88f819c83a52f",
  {
    input: {
      lr: 0.0004,
      xl: true,
      seed: 42,
      model: "stabilityai/stable-diffusion-xl-base-1.0",
      prompt: "a photo of aisudhca bag",
      xformers: false,
      num_steps: 2000,
      scheduler: "constant",
      batch_size: 1,
      num_cycles: 1,
      resolution: 1024,
      center_crop: false,
      warmup_steps: 0,
      use_8bit_adam: false,
      mixed_precision: "bf16",
      train_text_encoder: true,
      checkpointing_steps: 500,
      gradient_accumulation: 1
    }
  }
);

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

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

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

Add a payment method to run this model.

Sign in with GitHub

By signing in, you agree to our
terms of service and privacy policy

Output

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

Run time and cost

This model runs on Nvidia L40S GPU hardware. We don't yet have enough runs of this model to provide performance information.

Readme

AutoTrain Advanced: faster and easier training and deployments of state-of-the-art machine learning models.