diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py index 42aee4a84822..9a542bb2941c 100644 --- a/src/diffusers/loaders/single_file_utils.py +++ b/src/diffusers/loaders/single_file_utils.py @@ -2406,7 +2406,6 @@ def remove_keys_(key: str, state_dict): "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: