Readme
…
Model description
…
Intended use
…
Ethical considerations
…
Caveats and recommendations
…
Run this model in Node.js with one line of code:
npm install replicate
REPLICATE_API_TOKEN
environment variable:export REPLICATE_API_TOKEN=<paste-your-token-here>
Find your API token in your account settings.
import Replicate from "replicate";
import fs from "node:fs";
const replicate = new Replicate({
auth: process.env.REPLICATE_API_TOKEN,
});
Run anotherjesse/dreambooth-hf using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.
const output = await replicate.run(
"anotherjesse/dreambooth-hf:de0866d111b584905eacdf883b470a5ccffdc0d07701cca147d193334762adbe",
{
input: {
seed: 1337,
revision: "fp16",
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,
pretrained_model: "runwayml/stable-diffusion-v1-5",
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.
No output yet! Press "Submit" to start a prediction.
This model runs on Nvidia A100 (80GB) GPU hardware. We don't yet have enough runs of this model to provide performance information.
…
…
…
…
…