ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 11:11:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added experimental dfe 5

Browse Source
This commit is contained in:
Jaret Burkett
2025-09-21 10:48:52 -06:00
parent 20dfe1b4d5
commit 28728a1e92
2 changed files with 76 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -577,7 +577,7 @@ class SDTrainer(BaseSDTrainProcess):
dfe_loss += torch.nn.functional.mse_loss(pred_feature_list[i], target_feature_list[i], reduction="mean")
additional_loss += dfe_loss * self.train_config.diffusion_feature_extractor_weight * 100.0
elif self.dfe.version == 3 or self.dfe.version == 4:
elif self.dfe.version in [3, 4, 5]:
dfe_loss = self.dfe(
noise=noise,
noise_pred=noise_pred,