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

Update sdxl_train_util.py

Browse Source
This commit is contained in:
DKnight54
2025-03-11 00:23:23 +08:00
committed by GitHub
parent 8b8640149d
commit 34776180ca
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/sdxl_train_util.py
View File

@@ -94,7 +94,7 @@ def _load_target_model(
except ValueError as ex:
if variant is not None:
logger.info("try to load default model")
pipe = StableDiffusionXLPipeline.from_pretrained(name_or_path, variant=None, tokenizer=None)
pipe = StableDiffusionXLPipeline.from_pretrained(name_or_path, torch_dtype=model_dtype, variant=None, tokenizer=None)
else:
raise ex
except EnvironmentError as ex: