Update README.md

README.md

@@ -63,6 +63,9 @@ FlashSDXL can also be combined with existing LoRAs to unlock few steps generatio
 from diffusers import DiffusionPipeline, LCMScheduler
 import torch
 
+user_lora_id = "TheLastBen/Papercut_SDXL"
+trigger_word = "papercut"
+
 flash_lora_id = "jasperai/flash-sdxl"
 
 # Load Pipeline
@@ -78,14 +81,18 @@ pipe.scheduler = LCMScheduler.from_config(
 
 # Load LoRAs
 pipe.load_lora_weights(flash_lora_id, adapter_name="flash")
-pipe.load_lora_weights("TheLastBen/Papercut_SDXL", adapter_name="paper")
+pipe.load_lora_weights(user_lora_id, adapter_name="lora")
 
-pipe.set_adapters(["flash", "paper"], adapter_weights=[1.0, 1.0])
+pipe.set_adapters(["flash", "lora"], adapter_weights=[1.0, 1.0])
 pipe.to(device="cuda", dtype=torch.float16)
 
-prompt = "papercut, a cute corgi"
+prompt = f"{trigger_word} a cute corgi"
 
-image = pipe(prompt, num_inference_steps=4, guidance_scale=0).images[0]
+image = pipe(
+    prompt,
+    num_inference_steps=4,
+    guidance_scale=0
+).images[0]
 ```
 <p align="center">
    <img style="width:400px;" src="images/corgi.jpg">