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

Add stepped loss type

Browse Source
This commit is contained in:
Jaret Burkett
2025-09-22 15:50:12 -06:00
parent 28728a1e92
commit f74475161e
7 changed files with 108 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/models/diffusion_feature_extraction.py
View File

@@ -630,7 +630,7 @@ class DiffusionFeatureExtractor5(DiffusionFeatureExtractor4):
stepped_chunks.append(stepped)
# ---- Inverse-Gaussian recovery at the target timestep ----
t_01 = (scheduler.sigmas[target_idx] / 1000).to(stepped.device).to(stepped.dtype)
t_01 = (scheduler.sigmas[target_idx]).to(stepped.device).to(stepped.dtype)
original_samples = (stepped - t_01 * noise_i) / (1.0 - t_01)
x0_pred_chunks.append(original_samples)