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

Finished up first frame for i2v adapter

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-12 17:13:04 -06:00
parent cd37ccfc2e
commit 6fb44db6a0
5 changed files with 206 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -1046,10 +1046,11 @@ class SDTrainer(BaseSDTrainProcess):
quad_count = random.randint(1, 4)
self.adapter.train()
self.adapter.trigger_pre_te(
tensors_0_1=clip_images if not is_reg else None, # on regs we send none to get random noise
tensors_preprocessed=clip_images if not is_reg else None, # on regs we send none to get random noise
is_training=True,
has_been_preprocessed=True,
quad_count=quad_count,
batch_tensor=batch.tensor if not is_reg else None,
batch_size=noisy_latents.shape[0]
)