Fix previews for rf-inversion too

This commit is contained in:
kijai
2025-01-06 11:45:01 +02:00
parent 1d67d090da
commit 2eb34af146
+3 -3
View File
@@ -252,7 +252,7 @@ class HyVideoInverseSampler:
progress_bar.update()
if callback is not None:
callback(idx, latents.detach()[-1].permute(1,0,2,3), None, steps)
callback(idx, (latent_model_input - noise_pred).detach()[0].permute(1,0,2,3), None, steps)
else:
comfy_pbar.update(1)
@@ -458,7 +458,7 @@ class HyVideoReSampler:
progress_bar.update()
if callback is not None:
callback(idx, latents.detach()[-1].permute(1,0,2,3), None, steps)
callback(idx, (latent_model_input - noise_pred).detach()[0].permute(1,0,2,3), None, steps)
else:
comfy_pbar.update(1)
@@ -695,7 +695,7 @@ class HyVideoPromptMixSampler:
progress_bar.update()
if callback is not None:
callback(idx, latents.detach()[-1].permute(1,0,2,3), None, steps)
callback(idx, (latent_model_input - noise_pred).detach()[0].permute(1,0,2,3), None, steps)
else:
comfy_pbar.update(1)