mgonline2021 / drmb-a40l

  • Public
  • 0 runs
  • L40S
Iterate in playground

Input

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

npx create-replicate --model=mgonline2021/drmb-a40l
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 mgonline2021/drmb-a40l using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "mgonline2021/drmb-a40l:3287a348455da22b575718c3c2cd8718d38d6d919b7840ab73ed693c92a35f34",
  {
    input: {
      seed: 1337,
      scale_lr: false,
      adam_beta1: 0.9,
      adam_beta2: 0.999,
      pad_tokens: false,
      resolution: 512,
      center_crop: false,
      adam_epsilon: 1e-8,
      lr_scheduler: "constant",
      learning_rate: 0.000001,
      max_grad_norm: 1,
      n_save_sample: 4,
      use_8bit_adam: false,
      lr_warmup_steps: 0,
      max_train_steps: 2000,
      num_class_images: 50,
      num_train_epochs: 1,
      save_infer_steps: 50,
      train_batch_size: 1,
      adam_weight_decay: 0.01,
      prior_loss_weight: 1,
      sample_batch_size: 4,
      train_text_encoder: true,
      save_guidance_scale: 7.5,
      gradient_checkpointing: false,
      with_prior_preservation: true,
      gradient_accumulation_steps: 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.

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.