zelenioncode / dreambooth_sdxl

  • Public
  • 155 runs
  • A100 (80GB)
Iterate in playground

Input

Set the REPLICATE_API_TOKEN environment variable:
export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Run zelenioncode/dreambooth_sdxl 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": "zelenioncode/dreambooth_sdxl:e90767fffa851f984703defc193b1861d3b6d79acbcf1a798a3b9a1dac0bc176",
    "input": {
      "vae": "stabilityai/sdxl-vae",
      "input": "https://huggingface.co/WGlint/SafetensorsFromReplicate/resolve/main/input.zip",
      "gender": "woman",
      "xformers": true,
      "class_reg": "",
      "full_fp16": true,
      "model_sdxl": "Stable Diffusion XL",
      "name_model": "Dreambooth_sdxl",
      "resolution": "1024,1024",
      "no_half_vae": true,
      "warmup_init": false,
      "lr_scheduler": "constant",
      "mem_eff_attn": true,
      "noise_offset": 0,
      "repeat_input": 100,
      "weight_decay": 0.01,
      "cache_latents": true,
      "learning_rate": 0.00001,
      "max_grad_norm": 0,
      "relative_step": false,
      "use_class_reg": false,
      "optimizer_type": "AdaFactor",
      "save_precision": "fp16",
      "max_train_steps": 3000,
      "mixed_precision": "fp16",
      "scale_parameter": false,
      "repeat_class_reg": 1,
      "train_batch_size": 1,
      "bucket_no_upscale": true,
      "bucket_reso_steps": 64,
      "learning_rate_te1": 0.00001,
      "learning_rate_te2": 0.00001,
      "token_huggingface": "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",
      "folder_huggingface": "",
      "save_every_n_steps": 1,
      "train_text_encoder": true,
      "repo_id_huggingface": "WGlint/SafetensorsFromReplicate",
      "save_every_n_epochs": 1,
      "send_to_huggingface": false,
      "cache_latents_to_disk": true,
      "gradient_checkpointing": true,
      "learning_rate_te1_bool": 0.000003,
      "learning_rate_te2_bool": 0,
      "lr_scheduler_num_cycles": 1,
      "max_data_loader_n_workers": 0,
      "num_cpu_threads_per_process": 4
    }
  }' \
  https://api.replicate.com/v1/predictions

To learn more, take a look at Replicate’s HTTP API reference docs.

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.

Readme

This model doesn't have a readme.