alle guten dinge sind drei
This commit is contained in:
parent
e16862571d
commit
895af63be1
|
@ -576,10 +576,7 @@ class DiffusersHolder():
|
||||||
|
|
||||||
for i, t in enumerate(timesteps):
|
for i, t in enumerate(timesteps):
|
||||||
# Set the right starting latents
|
# Set the right starting latents
|
||||||
if i < idx_start:
|
if i == idx_start:
|
||||||
list_latents_out.append(None)
|
|
||||||
continue
|
|
||||||
elif i == idx_start:
|
|
||||||
latents = latents_start.clone()
|
latents = latents_start.clone()
|
||||||
|
|
||||||
# Mix latents for crossfeeding
|
# Mix latents for crossfeeding
|
||||||
|
@ -587,6 +584,11 @@ class DiffusersHolder():
|
||||||
latents_mixtarget = list_latents_mixing[i - 1].clone()
|
latents_mixtarget = list_latents_mixing[i - 1].clone()
|
||||||
latents = interpolate_spherical(latents, latents_mixtarget, list_mixing_coeffs[i])
|
latents = interpolate_spherical(latents, latents_mixtarget, list_mixing_coeffs[i])
|
||||||
|
|
||||||
|
# Write latents out and skip
|
||||||
|
if i < idx_start:
|
||||||
|
list_latents_out.append(None)
|
||||||
|
continue
|
||||||
|
|
||||||
# expand the latents if we are doing classifier free guidance
|
# expand the latents if we are doing classifier free guidance
|
||||||
latent_model_input = torch.cat([latents] * 2) if self.pipe.do_classifier_free_guidance else latents
|
latent_model_input = torch.cat([latents] * 2) if self.pipe.do_classifier_free_guidance else latents
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue