You're looking at a specific version of this model. Jump to the model overview.
replicate /dreambooth:a8ba568d
Input
Run this model in Node.js with one line of code:
npm install replicate
REPLICATE_API_TOKEN
environment variableexport REPLICATE_API_TOKEN=<paste-your-token-here>
Find your API token in your account settings.
import Replicate from "replicate";
const replicate = new Replicate({
auth: process.env.REPLICATE_API_TOKEN,
});
Run replicate/dreambooth using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.
const output = await replicate.run(
"replicate/dreambooth:a8ba568da0313951a6b311b43b1ea3bf9f2ef7b9fd97ed94cebd7ffd2da66654",
{
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
}
}
);
console.log(output);
To learn more, take a look at the guide on getting started with Node.js.
pip install replicate
REPLICATE_API_TOKEN
environment variableexport REPLICATE_API_TOKEN=<paste-your-token-here>
Find your API token in your account settings.
import replicate
Run replicate/dreambooth using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.
output = replicate.run(
"replicate/dreambooth:a8ba568da0313951a6b311b43b1ea3bf9f2ef7b9fd97ed94cebd7ffd2da66654",
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
}
)
print(output)
To learn more, take a look at the guide on getting started with Python.
REPLICATE_API_TOKEN
environment variableexport REPLICATE_API_TOKEN=<paste-your-token-here>
Find your API token in your account settings.
Run replicate/dreambooth using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.
curl -s -X POST \
-H "Authorization: Bearer $REPLICATE_API_TOKEN" \
-H "Content-Type: application/json" \
-H "Prefer: wait" \
-d $'{
"version": "a8ba568da0313951a6b311b43b1ea3bf9f2ef7b9fd97ed94cebd7ffd2da66654",
"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
}
}' \
https://api.replicate.com/v1/predictions
To learn more, take a look at Replicate’s HTTP API reference docs.
brew install cog
If you don’t have Homebrew, there are other installation options available.
Pull and run replicate/dreambooth using Cog (this will download the full model and run it in your local environment):
cog predict r8.im/replicate/dreambooth@sha256:a8ba568da0313951a6b311b43b1ea3bf9f2ef7b9fd97ed94cebd7ffd2da66654 \
-i 'seed=1337' \
-i 'scale_lr=false' \
-i 'adam_beta1=0.9' \
-i 'adam_beta2=0.999' \
-i 'pad_tokens=false' \
-i 'resolution=512' \
-i 'center_crop=false' \
-i 'adam_epsilon=1e-8' \
-i 'lr_scheduler="constant"' \
-i 'learning_rate=0.000001' \
-i 'max_grad_norm=1' \
-i 'n_save_sample=4' \
-i 'use_8bit_adam=false' \
-i 'lr_warmup_steps=0' \
-i 'max_train_steps=2000' \
-i 'num_class_images=50' \
-i 'num_train_epochs=1' \
-i 'save_infer_steps=50' \
-i 'train_batch_size=1' \
-i 'adam_weight_decay=0.01' \
-i 'prior_loss_weight=1' \
-i 'sample_batch_size=4' \
-i 'train_text_encoder=true' \
-i 'save_guidance_scale=7.5' \
-i 'gradient_checkpointing=false' \
-i 'with_prior_preservation=true' \
-i 'gradient_accumulation_steps=1'
To learn more, take a look at the Cog documentation.
Pull and run replicate/dreambooth using Docker (this will download the full model and run it in your local environment):
docker run -d -p 5000:5000 --gpus=all r8.im/replicate/dreambooth@sha256:a8ba568da0313951a6b311b43b1ea3bf9f2ef7b9fd97ed94cebd7ffd2da66654
curl -s -X POST \ -H "Content-Type: application/json" \ -d $'{ "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 } }' \ http://localhost:5000/predictions
Add a payment method to run this model.
Each run costs approximately $1.10. Alternatively, try out our featured models for free.
By signing in, you agree to our
terms of service and privacy policy
Output
No output yet! Press "Submit" to start a prediction.