From a02056c56674a2ba8e908c6f9dd190139286822c Mon Sep 17 00:00:00 2001
From: reid3333 <120766169+reid3333@users.noreply.github.com>
Date: Sat, 5 Aug 2023 17:47:43 +0900
Subject: [PATCH] fix: load may fail if symbolic link points to relative path

---
 library/train_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/library/train_util.py b/library/train_util.py
index 3a68bed9..d1405643 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -3125,7 +3125,7 @@ def prepare_dtype(args: argparse.Namespace):
 
 def _load_target_model(args: argparse.Namespace, weight_dtype, device="cpu"):
     name_or_path = args.pretrained_model_name_or_path
-    name_or_path = os.readlink(name_or_path) if os.path.islink(name_or_path) else name_or_path
+    name_or_path = os.path.realpath(name_or_path) if os.path.islink(name_or_path) else name_or_path
     load_stable_diffusion_format = os.path.isfile(name_or_path)  # determine SD or Diffusers
     if load_stable_diffusion_format:
         print(f"load StableDiffusion checkpoint: {name_or_path}")