ltx2.3 connectors loading

Author: prishajain1

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

@@ -352,7 +352,8 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
     params = state.to_pure_dict()
     state = dict(nnx.to_flat_state(state))
 
-    params = load_connectors_weights(config.pretrained_model_name_or_path, params, "cpu", subfolder="")
+    filename = "ltx-2.3-22b-dev.safetensors" if getattr(config, "model_name", "") == "ltx2.3" else None
+    params = load_connectors_weights(config.pretrained_model_name_or_path, params, "cpu", subfolder="", filename=filename)
     if hasattr(config, "weights_dtype"):
       params = jax.tree_util.tree_map(lambda x: x.astype(config.weights_dtype), params)