scheduler loading

prishajain1 · prishajain1 · commit ffed5edcd064 · 2026-04-09T14:19:56.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -614,7 +614,7 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
   def load_scheduler(cls, config: HyperParameters):
     max_logging.log("Loading Scheduler...")
     scheduler, _ = FlaxFlowMatchScheduler.from_pretrained(
-        config.pretrained_model_name_or_path,
+        "Lightricks/LTX-2",
         subfolder="scheduler",
     )
     return scheduler

Original file line number	Diff line number	Diff line change
`@@ -614,7 +614,7 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):`
`614`	`614`	`def load_scheduler(cls, config: HyperParameters):`
`615`	`615`	`max_logging.log("Loading Scheduler...")`
`616`	`616`	`scheduler, _ = FlaxFlowMatchScheduler.from_pretrained(`
`617`		`- config.pretrained_model_name_or_path,`
	`617`	`+ "Lightricks/LTX-2",`
`618`	`618`	`subfolder="scheduler",`
`619`	`619`	`)`
`620`	`620`	`return scheduler`