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
}
}
);
console.log(output);