update

DN6 · DN6 · commit 7d3bcd76e33b · 2025-04-02T20:37:44.000+02:00
diff --git 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:

Original file line number	Diff line number	Diff line change
`@@ -2406,7 +2406,6 @@ def remove_keys_(key: str, state_dict):`
`2406`	`2406`	`"per_channel_statistics.channel": remove_keys_,`
`2407`	`2407`	`"per_channel_statistics.mean-of-means": remove_keys_,`
`2408`	`2408`	`"per_channel_statistics.mean-of-stds": remove_keys_,`
`2409`		`- "timestep_scale_multiplier": remove_keys_,`
`2410`	`2409`	`}`
`2411`	`2410`
`2412`	`2411`	`if "vae.decoder.last_time_embedder.timestep_embedder.linear_1.weight" in converted_state_dict:`