prompthunt / cog-sdxl-training

  • Public
  • 7 runs
  • L40S
Iterate in playground

Input

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

npx create-replicate --model=prompthunt/cog-sdxl-training
or set up a project from scratch
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 prompthunt/cog-sdxl-training using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "prompthunt/cog-sdxl-training:bef5d92c784ae23bb8c5f79ceb4e58fb581c039513be7c4e2d60de349f132b6e",
  {
    input: {
      ti_lr: 0.0003,
      is_lora: true,
      lora_lr: 0.0001,
      verbose: true,
      lora_rank: 32,
      resolution: 768,
      lr_scheduler: "constant",
      token_string: "TOK",
      caption_prefix: "a photo of TOK, ",
      lr_warmup_steps: 100,
      max_train_steps: 1000,
      num_train_epochs: 4000,
      train_batch_size: 4,
      unet_learning_rate: 0.000001,
      checkpointing_steps: 999999,
      clipseg_temperature: 1,
      input_images_filetype: "infer",
      crop_based_on_salience: true,
      use_face_detection_instead: false
    }
  }
);

// 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.

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

This model doesn't have a readme.