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

Merge d65f46c297 into 1a3ec9ea74

Browse Source
This commit is contained in:
Daichi Nagashima
2026-03-13 02:46:23 +00:00
committed by GitHub
parent 1a3ec9ea74 d65f46c297
commit b9dc4f4b5c
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

@@ -83,7 +83,7 @@ def _load_target_model(
# Diffusers model is loaded to CPU
from diffusers import StableDiffusionXLPipeline
variant = "fp16" if weight_dtype == torch.float16 else None
variant = "fp16" if weight_dtype == torch.float16 else "bf16" if weight_dtype == torch.bfloat16 else None
logger.info(f"load Diffusers pretrained models: {name_or_path}, variant={variant}")
try:
try: