Readme
Please ignore this - use dreambooth api
this is where I test new dreambooths
pip 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
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.
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.
Please ignore this - use dreambooth api