anotherjesse / dreambooth

this is where I test new dreambooths

  • Public
  • 276.5K runs
  • A100 (80GB)
Iterate in playground

Input

pip 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 the client:
import replicate

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

output = replicate.run(
    "anotherjesse/dreambooth:926e61c5948d452a942e258d8a0a0f09a2c1749d2e2f153b3cccef6f18d6043f",
    input={
        "seed": 1337,
        "hflip": False,
        "lr_power": 1,
        "scale_lr": False,
        "adam_beta1": 0.9,
        "adam_beta2": 0.999,
        "resolution": 768,
        "center_crop": False,
        "adam_epsilon": 1e-8,
        "lr_scheduler": "constant",
        "learning_rate": 0.000005,
        "lr_num_cycles": 1,
        "max_grad_norm": 1,
        "use_8bit_adam": False,
        "lr_warmup_steps": 0,
        "max_train_steps": 2000,
        "num_class_images": 50,
        "num_train_epochs": 1,
        "pretrained_model": "stabilityai/stable-diffusion-2-1",
        "train_batch_size": 1,
        "adam_weight_decay": 0.01,
        "prior_loss_weight": 1,
        "sample_batch_size": 4,
        "train_text_encoder": True,
        "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.

Output

No output yet! Press "Submit" to start a prediction.

Run time and cost

This model runs on Nvidia A100 (80GB) GPU hardware. We don't yet have enough runs of this model to provide performance information.