update ui and add examples

2025-05-13 07:00:35 +00:00
parent 6db26bae5d
commit 6ca5bd880e
10 changed files with 258 additions and 119 deletions
@@ -604,6 +604,7 @@ class Pipeline(LightningModule):
        step = self.global_step
        checkpoint_name = f"epoch={epoch}-step={step}_lora"
        checkpoint_dir = os.path.join(log_dir, "checkpoints", checkpoint_name)
+        os.makedirs(checkpoint_dir, exist_ok=True)
        self.transformers.save_lora_adapter(checkpoint_dir, adapter_name=self.adapter_name)
        return state