File tree Expand file tree Collapse file tree 2 files changed +1
-1
lines changed
docs/source/en/api/pipelines/stable_diffusion
src/diffusers/models/autoencoders Expand file tree Collapse file tree 2 files changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -211,7 +211,6 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
211211 " stabilityai/stable-diffusion-3-medium-diffusers" , torch_dtype = torch.float16
212212)
213213pipe.vae = AutoencoderTiny.from_pretrained(" madebyollin/taesd3" , torch_dtype = torch.float16)
214- pipe.vae.config.shift_factor = 0.0
215214pipe = pipe.to(" cuda" )
216215
217216prompt = " slice of delicious New York-style berry cheesecake"
Original file line number Diff line number Diff line change @@ -111,6 +111,7 @@ def __init__(
111111 latent_shift : float = 0.5 ,
112112 force_upcast : bool = False ,
113113 scaling_factor : float = 1.0 ,
114+ shift_factor : float = 0.0 ,
114115 ):
115116 super ().__init__ ()
116117
You can’t perform that action at this time.
0 commit comments