test

Author: prishajain1

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

@@ -1223,6 +1223,7 @@ def __call__(
     with context_manager, axis_rules_context:
       connectors_graphdef, connectors_state = nnx.split(self.connectors)
 
+      @staticmethod
       @jax.jit
       def run_connectors(graphdef, state, hidden_states, attention_mask):
         model = nnx.merge(graphdef, state)