Luo-Yihong
commited on
Commit
•
7422ed6
1
Parent(s):
b9851ce
Update README.md
Browse files
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
|