m18coppola
commited on
Commit
•
6a35ea3
1
Parent(s):
a33ea52
Update README.md
Browse files
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 |
|