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

Update hidream defaults, pass additional information to flow guidance

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-16 13:03:04 -06:00
parent d5a64006b5
commit fecc64e646
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/guidance.py
View File

@@ -649,11 +649,13 @@ def targeted_flow_guidance(
noise,
timesteps
).detach()
unconditional_noisy_latents = sd.condition_noisy_latents(unconditional_noisy_latents, batch)
conditional_noisy_latents = sd.add_noise(
conditional_latents,
noise,
timesteps
).detach()
conditional_noisy_latents = sd.condition_noisy_latents(conditional_noisy_latents, batch)
# disable the lora to get a baseline prediction
sd.network.is_active = False