Update README.md
Browse files
README.md
CHANGED
@@ -84,7 +84,7 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
|
|
84 |
repo_id='lamm-mit/SD2x-leaf-inspired'
|
85 |
|
86 |
pipe = StableDiffusionPipeline.from_pretrained(repo_id,
|
87 |
-
scheduler = DPMSolverMultistepScheduler.from_pretrained(
|
88 |
torch_dtype=torch.float16,
|
89 |
).to("cuda")
|
90 |
|
|
|
84 |
repo_id='lamm-mit/SD2x-leaf-inspired'
|
85 |
|
86 |
pipe = StableDiffusionPipeline.from_pretrained(repo_id,
|
87 |
+
scheduler = DPMSolverMultistepScheduler.from_pretrained(repo_id, subfolder="scheduler"),
|
88 |
torch_dtype=torch.float16,
|
89 |
).to("cuda")
|
90 |
|