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

Changes to handle a different DFE arch

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-27 11:05:16 -06:00
parent fd13bd73a6
commit 1f541bc5d8
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -305,8 +305,11 @@ class SDTrainer(BaseSDTrainProcess):
# enable gradient checkpointing on the vae
if vae is not None and self.train_config.gradient_checkpointing:
vae.enable_gradient_checkpointing()
vae.train()
try:
vae.enable_gradient_checkpointing()
vae.train()
except:
pass
def process_output_for_turbo(self, pred, noisy_latents, timesteps, noise, batch):