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

Update lumina_train_network.py

Browse Source
This commit is contained in:
duongve13112002
2025-09-29 19:47:32 +07:00
committed by GitHub
parent f69a8f9e53
commit b32d66cfd2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lumina_train_network.py
View File

@@ -268,7 +268,7 @@ class LuminaNetworkTrainer(train_network.NetworkTrainer):
# NextDiT forward expects (x, t, cap_feats, cap_mask) # NextDiT forward expects (x, t, cap_feats, cap_mask)
model_pred = dit( model_pred = dit(
x=img, # image latents (B, C, H, W) x=img, # image latents (B, C, H, W)
t=timesteps / 1000, # timesteps需要除以1000来匹配模型预期 t= 1 - timesteps / 1000, # timesteps需要除以1000来匹配模型预期
cap_feats=gemma2_hidden_states, # Gemma2的hidden states作为caption features cap_feats=gemma2_hidden_states, # Gemma2的hidden states作为caption features
cap_mask=gemma2_attn_mask.to(dtype=torch.int32), # Gemma2的attention mask cap_mask=gemma2_attn_mask.to(dtype=torch.int32), # Gemma2的attention mask
) )