Update hidream defaults, pass additional information to flow guidance

This commit is contained in:
Jaret Burkett committed 2025-04-16 13:03:04 -06:00
1 parent d5a64006b5
commit fecc64e646
2 files changed
+12 -4

No files matched your search

+2
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