weight loading for tokenizer

prishajain1 · prishajain1 · commit 23bb98ef78d7 · 2026-04-09T11:30:11.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -333,7 +333,7 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
             }
         )
       connectors = LTX2AudioVideoGemmaTextEncoder.from_config(
-          config.pretrained_model_name_or_path,
+          "Lightricks/LTX-2",
           subfolder="connectors",
           rngs=rngs,
           mesh=mesh,

Original file line number	Diff line number	Diff line change
`@@ -333,7 +333,7 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):`
`333`	`333`	`}`
`334`	`334`	`)`
`335`	`335`	`connectors = LTX2AudioVideoGemmaTextEncoder.from_config(`
`336`		`- config.pretrained_model_name_or_path,`
	`336`	`+ "Lightricks/LTX-2",`
`337`	`337`	`subfolder="connectors",`
`338`	`338`	`rngs=rngs,`
`339`	`339`	`mesh=mesh,`