m18coppola commited on
Commit
6a35ea3
1 Parent(s): a33ea52

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-v2-lightning', torch_dtype=torch.float16, variant="fp16")
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