work on taesd encode

2026-09-18 16:54:33 +02:00 · 2023-10-31 18:41:14 -04:00
parent ec0d49acb7
commit 072b870b3f
4 changed files with 22 additions and 6 deletions
@@ -1123,6 +1123,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
                if shared.opts.sd_vae_sliced_encode and len(decoded_samples) > 1:
                    samples = torch.stack([self.sd_model.get_first_stage_encoding(self.sd_model.encode_first_stage(torch.unsqueeze(resized_sample, 0)))[0] for resized_sample in resized_samples])
                else:
+                    # TODO add TEASD support
                    samples = self.sd_model.get_first_stage_encoding(self.sd_model.encode_first_stage(resized_samples))
                image_conditioning = self.img2img_image_conditioning(resized_samples, samples)
            else: