stabilityai

/

stable-diffusion-3-medium-diffusers

like 326

how to do this?

First import the corresponding scheduler eg:

from diffusers import StableDiffusion3Pipeline, FlowMatchEulerDiscreteScheduler

and then use it in the pipeline eg:

....
pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
pipeline.scheduler = FlowMatchEulerDiscreteScheduler.from_config(pipeline.scheduler.config, shift=3.0)
pipeline = pipeline.to("cuda")
....