Skip to content

Commit 14a2282

Browse files
committed
update conversion script; todo: update 0.9.1 checkpoint with timestep_scale_multiplier
1 parent 3d3ad59 commit 14a2282

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

scripts/convert_ltx_to_diffusers.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -107,14 +107,6 @@ def remove_keys_(key: str, state_dict: Dict[str, Any]):
107107
"model.diffusion_model": remove_keys_,
108108
}
109109

110-
VAE_091_SPECIAL_KEYS_REMAP = {
111-
"timestep_scale_multiplier": remove_keys_,
112-
}
113-
114-
VAE_095_SPECIAL_KEYS_REMAP = {
115-
"timestep_scale_multiplier": remove_keys_,
116-
}
117-
118110

119111
def get_state_dict(saved_dict: Dict[str, Any]) -> Dict[str, Any]:
120112
state_dict = saved_dict
@@ -248,7 +240,6 @@ def get_vae_config(version: str) -> Dict[str, Any]:
248240
"decoder_causal": False,
249241
}
250242
VAE_KEYS_RENAME_DICT.update(VAE_091_RENAME_DICT)
251-
VAE_SPECIAL_KEYS_REMAP.update(VAE_091_SPECIAL_KEYS_REMAP)
252243
elif version == "0.9.5":
253244
config = {
254245
"in_channels": 3,
@@ -279,7 +270,6 @@ def get_vae_config(version: str) -> Dict[str, Any]:
279270
"decoder_causal": False,
280271
}
281272
VAE_KEYS_RENAME_DICT.update(VAE_095_RENAME_DICT)
282-
VAE_SPECIAL_KEYS_REMAP.update(VAE_095_SPECIAL_KEYS_REMAP)
283273
return config
284274

285275

0 commit comments

Comments
 (0)