Skip to content

Commit f631da3

Browse files
committed
del unused NNXCombinedTimestepTextProjEmbeddings and use NNXPixArtAlphaTextProjection
1 parent 53cfaae commit f631da3

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/models/ltx2/transformer_ltx2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222
from maxdiffusion.models.ltx2.attention_ltx2 import LTX2Attention, LTX2RotaryPosEmbed
2323
from maxdiffusion.models.attention_flax import NNXSimpleFeedForward
24-
from maxdiffusion.models.embeddings_flax import NNXPixArtAlphaCombinedTimestepSizeEmbeddings, NNXPixArtAlphaTextProjection, NNXCombinedTimestepTextProjEmbeddings
24+
from maxdiffusion.models.embeddings_flax import NNXPixArtAlphaCombinedTimestepSizeEmbeddings, NNXPixArtAlphaTextProjection
2525
from maxdiffusion.models.gradient_checkpoint import GradientCheckpointType
2626
from maxdiffusion.configuration_utils import ConfigMixin, register_to_config
2727
from maxdiffusion.common_types import BlockSizes

0 commit comments

Comments
 (0)