Update README.md
Browse files
README.md
CHANGED
|
@@ -31,6 +31,30 @@ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
|
|
| 31 |
|
| 32 |
You should use a photo of CAMPUPUSBIER beer to trigger the image generation.
|
| 33 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
## Download model
|
| 35 |
|
| 36 |
Weights for this model are available in Safetensors format.
|
|
|
|
| 31 |
|
| 32 |
You should use a photo of CAMPUPUSBIER beer to trigger the image generation.
|
| 33 |
|
| 34 |
+
## Training Settings
|
| 35 |
+
!accelerate launch train_dreambooth_lora_sdxl.py \
|
| 36 |
+
--pretrained_model_name_or_path="stabilityai/stable-diffusion-xl-base-1.0" \
|
| 37 |
+
--pretrained_vae_model_name_or_path="madebyollin/sdxl-vae-fp16-fix" \
|
| 38 |
+
--dataset_name="campusbiernew" \
|
| 39 |
+
--output_dir="campusbiernew_LoRA" \
|
| 40 |
+
--caption_column="prompt"\
|
| 41 |
+
--mixed_precision="fp16" \
|
| 42 |
+
--instance_prompt="a photo of [CB] beer" \
|
| 43 |
+
--resolution=1024 \
|
| 44 |
+
--train_batch_size=1 \
|
| 45 |
+
--gradient_accumulation_steps=3 \
|
| 46 |
+
--gradient_checkpointing \
|
| 47 |
+
--learning_rate=1e-4 \
|
| 48 |
+
--snr_gamma=5.0 \
|
| 49 |
+
--lr_scheduler="constant" \
|
| 50 |
+
--lr_warmup_steps=0 \
|
| 51 |
+
--mixed_precision="fp16" \
|
| 52 |
+
--use_8bit_adam \
|
| 53 |
+
--max_train_steps=500 \
|
| 54 |
+
--checkpointing_steps=717 \
|
| 55 |
+
--seed="0"
|
| 56 |
+
|
| 57 |
+
|
| 58 |
## Download model
|
| 59 |
|
| 60 |
Weights for this model are available in Safetensors format.
|