From 94b676d91382267e7429bd68362019868affd9d1 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Mon, 13 Feb 2023 17:19:18 +0100
Subject: Update

---
 training/strategy/lora.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'training/strategy/lora.py')

diff --git a/training/strategy/lora.py b/training/strategy/lora.py
index bc10e58..4dd1100 100644
--- a/training/strategy/lora.py
+++ b/training/strategy/lora.py
@@ -91,7 +91,7 @@ def lora_strategy_callbacks(
         print(f"Saving checkpoint for step {step}...")
 
         unet_ = accelerator.unwrap_model(unet)
-        unet_.save_attn_procs(checkpoint_output_dir.joinpath(f"{step}_{postfix}"))
+        unet_.save_attn_procs(checkpoint_output_dir / f"{step}_{postfix}")
         del unet_
 
     @torch.no_grad()
-- 
cgit v1.2.3-54-g00ecf