Update README.md

README.md

@@ -40,4 +40,17 @@ These are LoRA adaption weights for CompVis/stable-diffusion-v1-4. The weights w
 
 ## Training details
 
-[TODO: describe the data used to train the model]
+[TODO: describe the data used to train the model]
+
+!accelerate launch --mixed_precision="no" /content/diffusers/examples/text_to_image/train_text_to_image_lora.py \
+  --pretrained_model_name_or_path=$MODEL_NAME \
+  --dataset_name=$DATASET_NAME --caption_column="text" \
+  --resolution=512 --random_flip \
+  --train_batch_size=6 \
+  --num_train_epochs=50 --checkpointing_steps=450 \
+  --learning_rate=1e-04 --lr_scheduler="constant" --lr_warmup_steps=0 \
+  --seed=42 \
+  --output_dir="fine_tuning_gen" \
+  --validation_prompt "Iris of the eye, web pattern, blue and light blue, beautiful complex pattern" \
+  --report_to="wandb" \
+  --push_to_hub