training dreambooth LoRAs, with automatic checkpointing and resuming to s3-compatible storage
6.0K
| Variable Name | Default Value | Description |
|---|---|---|
| LOG_LEVEL | INFO | Log level for the application |
| MODEL_NAME | stabilityai/stable-diffusion-xl-base-1.0 | Name of the base model used |
| INSTANCE_DIR | /images | Directory where training images are stored |
| OUTPUT_DIR | /output | Directory for output files |
| VAE_PATH | madebyollin/sdxl-vae-fp16-fix | name of the VAE model |
| PROMPT | photo of timberdog | Prompt describing the concept being trained |
| DREAMBOOTH_SCRIPT | train_dreambooth_lora_sdxl.py | Dreambooth training script path |
| RESOLUTION | 1024 | Resolution of the images |
| MAX_TRAIN_STEPS | 500 | Total number of training steps |
| CHECKPOINTING_STEPS | 50 | Save a checkpoint after every N steps |
| CHECKPOINT_BUCKET_NAME | None | Name of the bucket for checkpoints (if applicable) |
| CHECKPOINT_BUCKET_PREFIX | None | Prefix for the checkpoint bucket (if applicable) |
| DATA_BUCKET_NAME | None | Name of the bucket for data storage (if applicable) |
| DATA_BUCKET_PREFIX | None | Prefix for the data bucket (if applicable) |
| WEBHOOK_URL | None | URL for webhooks |
| PROGRESS_WEBHOOK_URL | Value of 'WEBHOOK_URL' | URL for progress webhooks |
| COMPLETE_WEBHOOK_URL | Value of 'WEBHOOK_URL' | URL for complete webhooks |
| WEBHOOK_AUTH_HEADER | None | Authorization header for webhooks |
| PROGRESS_WEBHOOK_AUTH_HEADER | Value of 'WEBHOOK_AUTH_HEADER' | Authorization header for progress webhooks |
| COMPLETE_WEBHOOK_AUTH_HEADER | Value of 'WEBHOOK_AUTH_HEADER' | Authorization header for complete webhooks |
| WEBHOOK_AUTH_VALUE | None | Authorization value for webhooks |
| PROGRESS_WEBHOOK_AUTH_VALUE | Value of 'WEBHOOK_AUTH_VALUE' | Authorization value for progress webhooks |
| COMPLETE_WEBHOOK_AUTH_VALUE | Value of 'WEBHOOK_AUTH_VALUE' | Authorization value for complete webhooks |
| SALAD_MACHINE_ID | None | ID for the salad machine |
| SALAD_CONTAINER_GROUP_ID | None | ID for the salad container group |
Content type
Image
Digest
sha256:4ce6597db…
Size
7.5 GB
Last updated
over 2 years ago
docker pull saladtechnologies/dreambooth:kelpie-epicphotogasm