Update README.md
Browse files
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
|
| 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")
|