Unverified Commit cd308200 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Core] Add `shift_factor` to SD3 tiny autoencoder (#8618)

* shift factor argument to tiny

* remove shift factor rejigging from the sd3 docs
parent f3209b5b
...@@ -211,7 +211,6 @@ pipe = StableDiffusion3Pipeline.from_pretrained( ...@@ -211,7 +211,6 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
"stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16 "stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16
) )
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesd3", torch_dtype=torch.float16) pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesd3", torch_dtype=torch.float16)
pipe.vae.config.shift_factor = 0.0
pipe = pipe.to("cuda") pipe = pipe.to("cuda")
prompt = "slice of delicious New York-style berry cheesecake" prompt = "slice of delicious New York-style berry cheesecake"
......
...@@ -111,6 +111,7 @@ class AutoencoderTiny(ModelMixin, ConfigMixin): ...@@ -111,6 +111,7 @@ class AutoencoderTiny(ModelMixin, ConfigMixin):
latent_shift: float = 0.5, latent_shift: float = 0.5,
force_upcast: bool = False, force_upcast: bool = False,
scaling_factor: float = 1.0, scaling_factor: float = 1.0,
shift_factor: float = 0.0,
): ):
super().__init__() super().__init__()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment