import fix

Author: prishajain1

src/maxdiffusion/models/ltx2/text_encoders/feature_extractor_ltx2.py

@@ -14,7 +14,7 @@
 limitations under the License.
 """
 
-from typing import Tuple, Union
+from typing import Tuple, Union, Optional
 import jax.numpy as jnp
 from flax import nnx
 from maxdiffusion import common_types

src/maxdiffusion/models/ltx2/text_encoders/text_encoders_ltx2.py

@@ -14,7 +14,7 @@
 limitations under the License.
 """
 
-from typing import Tuple, Union, List
+from typing import Tuple, Union, List, Optional
 import jax
 import jax.numpy as jnp
 from flax import nnx