Skip to content

GitLab

Unverified Commit 6febc08b authored by Bernd Doser's avatar Bernd Doser Committed by GitHub
Browse files

Fix type of force_upcast to bool (#12046)

parent 9a2eaed0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/autoencoders/autoencoder_kl.py
View file @ 6febc08b
...@@ -90,7 +90,7 @@ class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalModelMixin, PeftAdapter ...@@ -90,7 +90,7 @@ class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalModelMixin, PeftAdapter
shift_factor: Optional[float] = None, shift_factor: Optional[float] = None,
latents_mean: Optional[Tuple[float]] = None, latents_mean: Optional[Tuple[float]] = None,
latents_std: Optional[Tuple[float]] = None, latents_std: Optional[Tuple[float]] = None,
force_upcast: float = True, force_upcast: bool = True,
use_quant_conv: bool = True, use_quant_conv: bool = True,
use_post_quant_conv: bool = True, use_post_quant_conv: bool = True,
mid_block_add_attention: bool = True, mid_block_add_attention: bool = True,
......
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