Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ pip install diffusers transformers accelerate
|
|
37 |
from diffusers import AutoPipelineForText2Image, LCMScheduler
|
38 |
import torch
|
39 |
|
40 |
-
pipe = AutoPipelineForText2Image.from_pretrained('lykon
|
41 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
42 |
pipe = pipe.to("cuda")
|
43 |
|
|
|
37 |
from diffusers import AutoPipelineForText2Image, LCMScheduler
|
38 |
import torch
|
39 |
|
40 |
+
pipe = AutoPipelineForText2Image.from_pretrained('lykon/dreamshaper-8-lcm', torch_dtype=torch.float16, variant="fp16")
|
41 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
42 |
pipe = pipe.to("cuda")
|
43 |
|