From 8b8640149d1a124115e46d8aaee69a29abf57b78 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Wed, 5 Mar 2025 01:30:18 +0800
Subject: [PATCH] Update sdxl_train_util.py

---
 library/sdxl_train_util.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/library/sdxl_train_util.py b/library/sdxl_train_util.py
index b74bea91..a7ae6b84 100644
--- a/library/sdxl_train_util.py
+++ b/library/sdxl_train_util.py
@@ -91,9 +91,9 @@ def _load_target_model(
                 pipe = StableDiffusionXLPipeline.from_pretrained(
                     name_or_path, torch_dtype=model_dtype, variant=variant, tokenizer=None
                 )
-            except EnvironmentError as ex:
+            except ValueError as ex:
                 if variant is not None:
-                    logger.info("try to load fp32 model")
+                    logger.info("try to load default model")
                     pipe = StableDiffusionXLPipeline.from_pretrained(name_or_path, variant=None, tokenizer=None)
                 else:
                     raise ex