fix

Author: prishajain1

src/maxdiffusion/checkpointing/ltx2_checkpointer.py

@@ -42,14 +42,18 @@ def __init__(self, config, checkpoint_type: str = LTX2_CHECKPOINT):
     self.opt_state = None
 
     self.checkpoint_manager: ocp.CheckpointManager = create_orbax_checkpoint_manager(
-        self.config.checkpoint_dir,
+        getattr(self.config, "checkpoint_dir", ""),
         enable_checkpointing=True,
         save_interval_steps=1,
         checkpoint_type=checkpoint_type,
         dataset_type=getattr(config, "dataset_type", None),
     )
 
   def load_ltx2_configs_from_orbax(self, step: Optional[int]) -> Tuple[Optional[dict], Optional[int]]:
+    if self.checkpoint_manager is None:
+      max_logging.log("No checkpoint manager configured, skipping Orbax load.")
+      return None, None
+      
     if step is None:
       step = self.checkpoint_manager.latest_step()
       max_logging.log(f"Latest LTX2 checkpoint step: {step}")