Update README.md

README.md

@@ -10,7 +10,7 @@ The YOSO was proposed in You Only Sample Once: Taming One-Step Text-To-Image Syn
 pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype = torch.float16)
 pipeline = pipeline.to('cuda')
 pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
-pipeline.load_lora_weights('Yihong666/yoso_sd1.5_lora')
+pipeline.load_lora_weights('Luo-Yihong/yoso_sd1.5_lora')
 generator = torch.manual_seed(318)
 steps = 1
 bs = 1
@@ -51,7 +51,7 @@ We note that a small CFG can be used to enhance the image quality.
 pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/realistic-vision-v51", torch_dtype = torch.float16)
 pipeline = pipeline.to('cuda')
 pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
-pipeline.load_lora_weights('Yihong666/yoso_sd1.5_lora')
+pipeline.load_lora_weights('Luo-Yihong/yoso_sd1.5_lora')
 generator = torch.manual_seed(318)
 steps = 2
 imgs= pipeline(prompt="A photo of a man, XT3",