technicalkiwi
/
latentblending
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

This commit is contained in:
Johannes Stelzer 2024-01-08 16:19:08 +01:00
parent 4f901b5294
commit 196f8bc09e
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
diffusers_holder.py
View File

@ -576,7 +576,11 @@ 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: # Write latents out and skip
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
@ -584,10 +588,6 @@ 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(latents)
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
@ -701,7 +701,7 @@ class DiffusersHolder():
list_latents_out.append(None) list_latents_out.append(None)
continue continue
# Set the right starting latents # Set the right starting latents
if i == idx_start: elif i == idx_start:
latents = latents_start.clone() latents = latents_start.clone()
# Mix latents for crossfeeding # Mix latents for crossfeeding