import error fix

Author: prishajain1

src/maxdiffusion/models/ltx2/ltx2_3_utils.py

@@ -11,12 +11,7 @@
     LTX_2_0_VIDEO_VAE_RENAME_DICT,
 )
 
-LTX_2_3_VIDEO_VAE_RENAME_DICT = {
-    **LTX_2_0_VIDEO_VAE_RENAME_DICT,
-    # Decoder extra blocks
-    "up_blocks.7": "up_blocks.3.upsamplers.0",
-    "up_blocks.8": "up_blocks.3",
-}
+
 
 LTX_2_3_CONNECTORS_KEYS_RENAME_DICT = {
     "connectors.": "",

src/maxdiffusion/models/ltx2/ltx2_utils.py

@@ -24,7 +24,6 @@
 from safetensors import safe_open
 from flax.traverse_util import unflatten_dict, flatten_dict
 from ..modeling_flax_pytorch_utils import (rename_key, rename_key_and_reshape_tensor, torch2jax, validate_flax_state_dict)
-from .ltx2_3_utils import LTX_2_3_VIDEO_VAE_RENAME_DICT
 
 
 LTX_2_0_VIDEO_VAE_RENAME_DICT = {
@@ -55,6 +54,13 @@
     "per_channel_statistics.std-of-means": "latents_std",
 }
 
+LTX_2_3_VIDEO_VAE_RENAME_DICT = {
+    **LTX_2_0_VIDEO_VAE_RENAME_DICT,
+    # Decoder extra blocks
+    "up_blocks.7": "up_blocks.3.upsamplers.0",
+    "up_blocks.8": "up_blocks.3",
+}
+