import error fix

prishajain1 · prishajain1 · commit 8cf8a80b3c30 · 2026-04-09T12:13:02.000+05:30
diff --git a/src/maxdiffusion/models/ltx2/ltx2_3_utils.py b/src/maxdiffusion/models/ltx2/ltx2_3_utils.py
@@ -4,7 +4,7 @@
 from flax import nnx
 from flax.traverse_util import unflatten_dict, flatten_dict
 from maxdiffusion import max_logging
-from maxdiffusion.models.flax_pytorch_utils import (
+from ..modeling_flax_pytorch_utils import (
     load_sharded_checkpoint,
     validate_flax_state_dict,
 )

Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@`
`4`	`4`	`from flax import nnx`
`5`	`5`	`from flax.traverse_util import unflatten_dict, flatten_dict`
`6`	`6`	`from maxdiffusion import max_logging`
`7`		`-from maxdiffusion.models.flax_pytorch_utils import (`
	`7`	`+from ..modeling_flax_pytorch_utils import (`
`8`	`8`	`load_sharded_checkpoint,`
`9`	`9`	`validate_flax_state_dict,`
`10`	`10`	`)`