ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP more work on cogview4

Browse Source
This commit is contained in:
Jaret Burkett
2025-03-05 09:43:00 -07:00
parent 6f6fb90812
commit aa44828c0c
3 changed files with 51 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/samplers/custom_flowmatch_sampler.py
View File

@@ -89,7 +89,7 @@ class CustomFlowMatchEulerDiscreteScheduler(FlowMatchEulerDiscreteScheduler):
) -> torch.Tensor:
t_01 = (timesteps / 1000).to(original_samples.device)
# forward ODE
noisy_model_input = (1 - t_01) * original_samples + t_01 * noise
noisy_model_input = (1.0 - t_01) * original_samples + t_01 * noise
# reverse ODE
# noisy_model_input = (1 - t_01) * noise + t_01 * original_samples
return noisy_model_input