zhouzhengjun/lora_train

Public
5 runs

Input

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

npx create-replicate --model=zhouzhengjun/lora_train
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";
import fs from "node:fs";

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

Run zhouzhengjun/lora_train using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "zhouzhengjun/lora_train:490c2d30a2996a69f617ea9a8e5fd17d31bbc032291b2097c324ac1670abb851",
  {
    input: {
      seed: 1337,
      scale_lr: true,
      lora_rank: 4,
      lora_scale: 1,
      resolution: 512,
      color_jitter: true,
      lr_scheduler: "constant",
      use_template: "object",
      clip_ti_decay: true,
      lora_dropout_p: 0.1,
      lr_warmup_steps: 0,
      weight_decay_ti: 0,
      learning_rate_ti: 0.0005,
      train_batch_size: 1,
      lr_scheduler_lora: "constant",
      weight_decay_lora: 0.001,
      continue_inversion: false,
      learning_rate_text: 0.00001,
      learning_rate_unet: 0.0001,
      max_train_steps_ti: 500,
      placeholder_tokens: "<s1>|<s2>",
      train_text_encoder: true,
      lr_warmup_steps_lora: 0,
      continue_inversion_lr: 0.0001,
      gradient_checkpointing: false,
      max_train_steps_tuning: 1000,
      gradient_accumulation_steps: 4,
      use_face_segmentation_condition: 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 T4 GPU hardware. We don't yet have enough runs of this model to provide performance information.

Readme

This model doesn't have a readme.