ehristoforu commited on
Commit
e34c99f
1 Parent(s): 299657c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -53,7 +53,7 @@ import torch
53
  from diffusers import StableDiffusionXLPipeline, DPMSolverSinglestepScheduler
54
 
55
  # Load model.
56
- pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", torch_dtype=torch.float16, variant="fp16").to("cuda")
57
 
58
  # Ensure sampler uses "trailing" timesteps.
59
  pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
 
53
  from diffusers import StableDiffusionXLPipeline, DPMSolverSinglestepScheduler
54
 
55
  # Load model.
56
+ pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", torch_dtype=torch.float16).to("cuda")
57
 
58
  # Ensure sampler uses "trailing" timesteps.
59
  pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")