Fix Single File loading for LTX VAE (#11200)

update

Fix Single File loading for LTX VAE (#11200)
update
aabf8ce2 · Dhruv Nair · GitHub · f10775b1 · aabf8ce2
Unverified Commit aabf8ce2 authored Apr 04, 2025 by Dhruv Nair Committed by GitHub Apr 04, 2025
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

src/diffusers/loaders/single_file_utils.py src/diffusers/loaders/single_file_utils.py +0 -1

No files found.
--- a/src/diffusers/loaders/single_file_utils.py
+++ b/src/diffusers/loaders/single_file_utils.py
@@ -2407,7 +2407,6 @@ def convert_ltx_vae_checkpoint_to_diffusers(checkpoint, **kwargs):
        "per_channel_statistics.channel": remove_keys_,
        "per_channel_statistics.mean-of-means": remove_keys_,
        "per_channel_statistics.mean-of-stds": remove_keys_,
-        "timestep_scale_multiplier": remove_keys_,
    }

    if "vae.decoder.last_time_embedder.timestep_embedder.linear_1.weight" in converted_state_dict: