harpomaxx commited on
Commit
c1e1c3d
1 Parent(s): 92e71a7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
README.md CHANGED
@@ -18,6 +18,30 @@ Training steps included:
18
 
19
  - train-text-encoder fine tuning
20
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
21
  ### 🧨 Diffusers
22
 
23
  This model can be used just like any other Stable Diffusion model. For more information,
 
18
 
19
  - train-text-encoder fine tuning
20
 
21
+
22
+ ```
23
+ accelerate launch train_dreambooth.py \
24
+ --pretrained_model_name_or_path=$MODEL_NAMEi \
25
+ --instance_data_dir=$INSTANCE_DIR \
26
+ --class_data_dir=$CLASS_DIR \
27
+ --output_dir=$OUTPUT_DIR \
28
+ --instance_prompt="sks style" \
29
+ --resolution=512 \
30
+ --train_batch_size=1 \
31
+ --gradient_accumulation_steps=1 \
32
+ --gradient_checkpointing \
33
+ --use_8bit_adam \
34
+ --enable_xformers_memory_efficient_attention \
35
+ --set_grads_to_none \
36
+ --learning_rate=1e-6 \
37
+ --lr_scheduler="constant" \
38
+ --lr_warmup_steps=0 \
39
+ --max_train_steps=2000 \
40
+ --train_text_encoder \
41
+ --mixed_precision=fp16
42
+ ```
43
+
44
+
45
  ### 🧨 Diffusers
46
 
47
  This model can be used just like any other Stable Diffusion model. For more information,