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

Update sdxl_train_util.py

Browse Source 
Try force stay FP16 on text encoders
This commit is contained in:
DKnight54
2025-01-23 23:41:49 +08:00
committed by GitHub
parent 6e3c1d0b58
commit fd3a14c9d7
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