Skip to content

Commit cd30820

Browse files
authored
[Core] Add shift_factor to SD3 tiny autoencoder (#8618)
* shift factor argument to tiny * remove shift factor rejigging from the sd3 docs
1 parent f3209b5 commit cd30820

File tree

2 files changed

+1
-1
lines changed

2 files changed

+1
-1
lines changed

docs/source/en/api/pipelines/stable_diffusion/stable_diffusion_3.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,6 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
211211
"stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16
212212
)
213213
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesd3", torch_dtype=torch.float16)
214-
pipe.vae.config.shift_factor = 0.0
215214
pipe = pipe.to("cuda")
216215

217216
prompt = "slice of delicious New York-style berry cheesecake"

src/diffusers/models/autoencoders/autoencoder_tiny.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)