Update README.md

README.md

@@ -40,4 +40,25 @@ These are LoRA adaption weights for stabilityai/stable-diffusion-2. The weights
 
 ## Training details
 
-[TODO: describe the data used to train the model]
+fine tuned using script>
+``` bash
+! accelerate launch    /kaggle/working/diffusers/examples/text_to_image/train_text_to_image_lora.py \
+  --pretrained_model_name_or_path="stabilityai/stable-diffusion-2" \
+  --dataset_name="ButterChicken98/plantvillage-image-text-pairs" \
+  --dataloader_num_workers=8 \
+  --resolution=256  --random_flip \
+  --train_batch_size=16 \
+  --gradient_accumulation_steps=4 \
+  --max_train_steps=3000 \
+  --learning_rate=1e-04 \
+  --max_grad_norm=1 \
+  --lr_scheduler="cosine" --lr_warmup_steps=0 \
+  --output_dir="/kaggle/working" \
+  --push_to_hub \
+  --caption_column="caption" \
+  --image_column="image" \
+  --hub_model_id="plantVillage-stableDiffusion-2-iter2_with_one_caption" \
+  --checkpointing_steps=1000 \
+  --validation_prompt="Tomato YellowLeaf Curl Virus" \
+  --seed=1337
+```