import fix in ltx_2_transformer_test.py

Author: prishajain1

src/maxdiffusion/tests/ltx_2_transformer_test.py

@@ -23,8 +23,8 @@
 from jax.sharding import Mesh
 from flax.linen import partitioning as nn_partitioning
 from maxdiffusion import pyconfig
-from ..max_utils import create_device_mesh
-from ..models.ltx2.transformer_ltx2 import (
+from maxdiffusion.max_utils import create_device_mesh
+from maxdiffusion.models.ltx2.transformer_ltx2 import (
     LTX2VideoTransformerBlock,
     LTX2VideoTransformer3DModel,
     LTX2AdaLayerNormSingle,