sdxl flag fix

This commit is contained in:
Johannes Stelzer 2024-01-09 14:51:27 +01:00
parent b1d9519a10
commit 4d4125e32b
1 changed files with 8 additions and 18 deletions

View File

@ -562,25 +562,15 @@ class LatentBlending():
self.dh.set_num_inference_steps(self.num_inference_steps) self.dh.set_num_inference_steps(self.num_inference_steps)
assert type(list_conditionings) is list, "list_conditionings need to be a list" assert type(list_conditionings) is list, "list_conditionings need to be a list"
if self.dh.use_sd_xl: text_embeddings = list_conditionings[0]
text_embeddings = list_conditionings[0] return self.dh.run_diffusion_sd_xl(
return self.dh.run_diffusion_sd_xl( text_embeddings=text_embeddings,
text_embeddings=text_embeddings, latents_start=latents_start,
latents_start=latents_start, idx_start=idx_start,
idx_start=idx_start, list_latents_mixing=list_latents_mixing,
list_latents_mixing=list_latents_mixing, mixing_coeffs=mixing_coeffs,
mixing_coeffs=mixing_coeffs, return_image=return_image)
return_image=return_image)
else:
text_embeddings = list_conditionings[0]
return self.dh.run_diffusion_standard(
text_embeddings=text_embeddings,
latents_start=latents_start,
idx_start=idx_start,
list_latents_mixing=list_latents_mixing,
mixing_coeffs=mixing_coeffs,
return_image=return_image)