We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 57b61de commit 0eb09cfCopy full SHA for 0eb09cf
1 file changed
src/maxdiffusion/models/ltx2/ltx2_3_utils.py
@@ -3,7 +3,7 @@
3
import numpy as np
4
from flax import nnx
5
from flax.traverse_util import unflatten_dict, flatten_dict
6
-from maxdiffusion.utils import max_logging
+from maxdiffusion import max_logging
7
from maxdiffusion.models.flax_pytorch_utils import (
8
load_sharded_checkpoint,
9
validate_flax_state_dict,
0 commit comments