Fix type of force_upcast to bool (#12046)

6febc08b · Bernd Doser · GitHub · 9a2eaed0 · 6febc08b
Unverified Commit 6febc08b authored Aug 02, 2025 by Bernd Doser Committed by GitHub Aug 02, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/models/autoencoders/autoencoder_kl.py src/diffusers/models/autoencoders/autoencoder_kl.py +1 -1

No files found.
--- a/src/diffusers/models/autoencoders/autoencoder_kl.py
+++ b/src/diffusers/models/autoencoders/autoencoder_kl.py
@@ -90,7 +90,7 @@ class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalModelMixin, PeftAdapter
        shift_factor: Optional[float] = None,
        latents_mean: Optional[Tuple[float]] = None,
        latents_std: Optional[Tuple[float]] = None,
-        force_upcast: float = True,
+        force_upcast: bool = True,
        use_quant_conv: bool = True,
        use_post_quant_conv: bool = True,
        mid_block_add_attention: bool = True,