diff --git a/diffusers_holder.py b/diffusers_holder.py index ef36193..37795d5 100644 --- a/diffusers_holder.py +++ b/diffusers_holder.py @@ -586,7 +586,7 @@ class DiffusersHolder(): # Write latents out and skip if i < idx_start: - list_latents_out.append(None) + list_latents_out.append(latents) continue # expand the latents if we are doing classifier free guidance