From 0ec774343672c4d90a913a7a1505dfe9cd642f4a Mon Sep 17 00:00:00 2001 From: Kohya S Date: Mon, 15 May 2023 20:57:53 +0900 Subject: [PATCH] show loading model path --- library/train_util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index 2a55a446..70466cda 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -2918,11 +2918,11 @@ def _load_target_model(args: argparse.Namespace, weight_dtype, device="cpu"): name_or_path = os.readlink(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("load StableDiffusion checkpoint") + print(f"load StableDiffusion checkpoint: {name_or_path}") text_encoder, vae, unet = model_util.load_models_from_stable_diffusion_checkpoint(args.v2, name_or_path, device) else: # Diffusers model is loaded to CPU - print("load Diffusers pretrained models") + print(f"load Diffusers pretrained models: {name_or_path}") try: pipe = StableDiffusionPipeline.from_pretrained(name_or_path, tokenizer=None, safety_checker=None) except EnvironmentError as ex: