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(
211
211
" stabilityai/stable-diffusion-3-medium-diffusers" , torch_dtype = torch.float16
212
212
)
213
213
pipe.vae = AutoencoderTiny.from_pretrained(" madebyollin/taesd3" , torch_dtype = torch.float16)
214
- pipe.vae.config.shift_factor = 0.0
215
214
pipe = pipe.to(" cuda" )
216
215
217
216
prompt = " slice of delicious New York-style berry cheesecake"
Original file line number Diff line number Diff line change @@ -111,6 +111,7 @@ def __init__(
111
111
latent_shift : float = 0.5 ,
112
112
force_upcast : bool = False ,
113
113
scaling_factor : float = 1.0 ,
114
+ shift_factor : float = 0.0 ,
114
115
):
115
116
super ().__init__ ()
116
117
You can’t perform that action at this time.
0 commit comments