Update sdxl_train_util.py

2026-04-15 08:36:41 +00:00 · 2025-03-05 01:30:18 +08:00
parent 9b9d20515a
commit 8b8640149d
1 changed files with 2 additions and 2 deletions
--- 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