VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: load may fail if symbolic link points to relative path

Browse Source
This commit is contained in:
reid3333
2023-08-05 17:47:43 +09:00
parent cf8021020f
commit a02056c566
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/train_util.py
View File

@@ -3125,7 +3125,7 @@ def prepare_dtype(args: argparse.Namespace):
def _load_target_model(args: argparse.Namespace, weight_dtype, device="cpu"): def _load_target_model(args: argparse.Namespace, weight_dtype, device="cpu"):
name_or_path = args.pretrained_model_name_or_path 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 load_stable_diffusion_format = os.path.isfile(name_or_path) # determine SD or Diffusers
if load_stable_diffusion_format: if load_stable_diffusion_format:
print(f"load StableDiffusion checkpoint: {name_or_path}") print(f"load StableDiffusion checkpoint: {name_or_path}")