weight loading for tokenizer

prishajain1 · prishajain1 · commit 1d0ae416368e · 2026-04-09T11:03:48.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -298,7 +298,7 @@ def disable_vae_tiling(self):
   def load_tokenizer(cls, config: HyperParameters):
     max_logging.log("Loading Gemma Tokenizer...")
     tokenizer = AutoTokenizer.from_pretrained(
-        config.pretrained_model_name_or_path,
+        "Lightricks/LTX-2",
         subfolder="tokenizer",
     )
     return tokenizer

Original file line number	Diff line number	Diff line change
`@@ -298,7 +298,7 @@ def disable_vae_tiling(self):`
`298`	`298`	`def load_tokenizer(cls, config: HyperParameters):`
`299`	`299`	`max_logging.log("Loading Gemma Tokenizer...")`
`300`	`300`	`tokenizer = AutoTokenizer.from_pretrained(`
`301`		`- config.pretrained_model_name_or_path,`
	`301`	`+ "Lightricks/LTX-2",`
`302`	`302`	`subfolder="tokenizer",`
`303`	`303`	`)`
`304`	`304`	`return tokenizer`