Update README.md

README.md

@@ -81,50 +81,41 @@ This LoRA enhances FLUX.1 [dev]'s ability to generate detailed autumn landsc
 First, ensure you have FLUX.1 [dev] installed:
 
 ```python
+# Install
 pip install -U diffusers
-Usage
-Loading with Diffusers
-Here's how to apply the LoRA to FLUX.1 [dev]:
 
+# Loading with Diffusers
 import torch
 from diffusers import FluxPipeline
 
 # Load base model
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload()
 
 # Load and apply LoRA weights
-pipe.load_lora_weights("Borcherding/seasonalLandsFluxDev-lora")
+pipe.load_lora_weights("Borcherding/seasonalLandsFluxDev-lora")
 
 # Generate image
-prompt = "autumnProxy A majestic maple tree with vibrant red and orange leaves, golden afternoon light"
-image = pipe(
+prompt = "autumnProxy A majestic maple tree with vibrant red and orange leaves, golden afternoon light"
+image = pipe(
     prompt,
-    height=1024,
-    width=1024,
-    guidance_scale=3.5,
-    num_inference_steps=50,
-    max_sequence_length=512,
-    generator=torch.Generator("cpu").manual_seed(0)
+    height=1024,
+    width=1024,
+    guidance_scale=3.5,
+    num_inference_steps=50,
+    max_sequence_length=512,
+    generator=torch.Generator("cpu").manual_seed(0)
 ).images[0]
-image.save("autumn-scene.png")
+image.save("autumn-scene.png")
 
 # Unload LoRA weights if needed
 pipe.unload_lora_weights()
-Merging LoRA (Optional)
-To permanently merge the LoRA weights with the base model:
-pythonCopyimport torch
-from diffusers import FluxPipeline
-
-# Load base model
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 
-# Load and merge LoRA weights
-pipe.load_lora_weights("Borcherding/SeasonalLandsFluxDev-lora")
+# Optional: Merge LoRA weights permanently
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("Borcherding/SeasonalLandsFluxDev-lora")
 pipe.merge_lora_weights()
-
-# Save merged model
-pipe.save_pretrained("seasonal-lands-merged")
+pipe.save_pretrained("seasonal-lands-merged")
 ```
 
 # Trigger Word