Luo-Yihong commited on
Commit
7422ed6
1 Parent(s): b9851ce

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -83,6 +83,7 @@ import torch
83
  from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
84
  pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/realistic-vision-v51", torch_dtype = torch.float16)
85
  pipeline = pipeline.to('cuda')
 
86
  pipeline.scheduler = DPMSolverMultistepScheduler.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="scheduler")
87
  generator = torch.manual_seed(323)
88
  steps = 2
 
83
  from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
84
  pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/realistic-vision-v51", torch_dtype = torch.float16)
85
  pipeline = pipeline.to('cuda')
86
+ pipeline.load_lora_weights('Luo-Yihong/yoso_sd1.5_lora')
87
  pipeline.scheduler = DPMSolverMultistepScheduler.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="scheduler")
88
  generator = torch.manual_seed(323)
89
  steps = 2