fix

Author: prishajain1

src/maxdiffusion/models/ltx2/transformer_ltx2.py

@@ -116,7 +116,6 @@ def __init__(
         epsilon=self.norm_eps,
         use_scale=self.norm_elementwise_affine,
         rngs=rngs,
-        rngs=rngs,
         dtype=jnp.float32,
         param_dtype=jnp.float32,
     )
@@ -140,7 +139,6 @@ def __init__(
         epsilon=self.norm_eps,
         use_scale=self.norm_elementwise_affine,
         rngs=rngs,
-        rngs=rngs,
         dtype=jnp.float32,
         param_dtype=jnp.float32,
     )
@@ -165,7 +163,6 @@ def __init__(
         epsilon=self.norm_eps,
         use_scale=self.norm_elementwise_affine,
         rngs=rngs,
-        rngs=rngs,
         dtype=jnp.float32,
         param_dtype=jnp.float32,
     )
@@ -190,7 +187,6 @@ def __init__(
         epsilon=self.norm_eps,
         use_scale=self.norm_elementwise_affine,
         rngs=rngs,
-        rngs=rngs,
         dtype=jnp.float32,
         param_dtype=jnp.float32,
     )
@@ -235,7 +231,6 @@ def __init__(
         epsilon=self.norm_eps,
         use_scale=self.norm_elementwise_affine,
         rngs=rngs,
-        rngs=rngs,
         dtype=jnp.float32,
         param_dtype=jnp.float32,
     )