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.
|