Update README.md

README.md

@@ -10,6 +10,8 @@ tags:
 - diffusers
 - diffusers-training
 - lora
+datasets:
+- lambdalabs/naruto-blip-captions
 ---
 
 <!-- This model card has been generated automatically according to the information the training script had access to. You
@@ -40,4 +42,45 @@ These are LoRA adaption weights for stable-diffusion-v1-5/stable-diffusion-v1-5.
 
 ## Training details
 
-[TODO: describe the data used to train the model]
+# Training Details - Stable Diffusion LoRA
+
+# Dataset
+# ------------------------------------------
+# The model was trained using the 'lambdalabs/naruto-blip-captions' dataset.
+# This dataset consists of Naruto character images with BLIP-generated captions.
+# It provides a diverse set of characters, poses, and backgrounds, 
+# making it suitable for fine-tuning Stable Diffusion on anime-style images.
+
+# Model
+# ------------------------------------------
+# Base Model: Stable Diffusion v1.5 (stable-diffusion-v1-5/stable-diffusion-v1-5)
+# Fine-tuning Method: LoRA (Low-Rank Adaptation)
+# Purpose: Specializing Stable Diffusion to generate Naruto-style anime characters.
+
+# Preprocessing
+# ------------------------------------------
+# - Images were resized to 512x512 resolution.
+# - Center cropping was applied to maintain aspect ratio.
+# - Random flipping was used as a data augmentation technique.
+
+# Training Configuration
+# ------------------------------------------
+# Batch Size: 1
+# Gradient Accumulation Steps: 4  # Simulates a larger batch size
+# Gradient Checkpointing: Enabled  # Reduces memory consumption
+# Max Training Steps: 800
+# Learning Rate: 1e-5 (constant schedule, no warmup)
+# Max Gradient Norm: 1  # Prevents gradient explosion
+# Memory Optimization: xFormers enabled for efficient attention computation
+
+# Validation
+# ------------------------------------------
+# - A validation prompt "A Naruto character" was used.
+# - 4 validation images were generated during training.
+# - Model checkpoints were saved every 500 steps.
+
+# Model Output
+# ------------------------------------------
+# - The fine-tuned LoRA model was saved to "sd-naruto-model".
+# - The model was pushed to the Hugging Face Hub:
+#   Repository: Bhaskar009/SD_1.5_LoRA