erikhsos commited on
Commit
0f0451e
1 Parent(s): 0f4f4e2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
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.