missing import

Author: prishajain1

src/maxdiffusion/models/ltx2/vocoder_ltx2.py

@@ -15,7 +15,7 @@
 """
 
 import math
-from typing import Sequence
+from typing import Optional, Sequence
 
 import jax
 import jax.numpy as jnp