Commit
•
561086c
1
Parent(s):
a33ea52
Update README.md (#5)
Browse files- Update README.md (6a35ea35e9d4023dad92198a4f21542dc7182e2a)
Co-authored-by: Michael Coppola <[email protected]>
README.md
CHANGED
|
@@ -41,7 +41,7 @@ pip install diffusers transformers accelerate
|
|
| 41 |
from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
|
| 42 |
import torch
|
| 43 |
|
| 44 |
-
pipe = AutoPipelineForText2Image.from_pretrained('lykon/dreamshaper-xl-
|
| 45 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
| 46 |
pipe = pipe.to("cuda")
|
| 47 |
|
|
|
|
| 41 |
from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
|
| 42 |
import torch
|
| 43 |
|
| 44 |
+
pipe = AutoPipelineForText2Image.from_pretrained('lykon/dreamshaper-xl-lightning', torch_dtype=torch.float16, variant="fp16")
|
| 45 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
| 46 |
pipe = pipe.to("cuda")
|
| 47 |
|