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

Update sdxl_train_util.py

Browse Source
This commit is contained in:
DKnight54
2025-01-24 00:19:09 +08:00
committed by GitHub
parent 68c65e0ae5
commit 8daa8b3283
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
library/sdxl_train_util.py
View File

@@ -107,10 +107,10 @@ def _load_target_model(
text_encoder2 = pipe.text_encoder_2
# convert to fp32 for cache text_encoders outputs
#if text_encoder1.dtype != torch.float32:
# text_encoder1 = text_encoder1.to(dtype=torch.float32)
#if text_encoder2.dtype != torch.float32:
# text_encoder2 = text_encoder2.to(dtype=torch.float32)
if text_encoder1.dtype != torch.float32:
text_encoder1 = text_encoder1.to(dtype=torch.float32)
if text_encoder2.dtype != torch.float32:
text_encoder2 = text_encoder2.to(dtype=torch.float32)
vae = pipe.vae
unet = pipe.unet