From 8e1cae5d668fb809f4470c681562e20aeffceda2 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 17 Feb 2023 00:00:27 +0200 Subject: [PATCH] Revert "[Pix2Pix0] Add utility function to get edit vector" (#2384) Revert "[Pix2Pix0] Add utility function to get edit vector (#2383)" This reverts commit 857c04cfbafcc43cbf04e647d7cd1699390af32a. --- .../pipeline_stable_diffusion_pix2pix_zero.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py index 42875d6690..af0a7e2a4b 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py @@ -542,20 +542,6 @@ class StableDiffusionPix2PixZeroPipeline(DiffusionPipeline): """Constructs the edit direction to steer the image generation process semantically.""" return (embs_target.mean(0) - embs_source.mean(0)).unsqueeze(0) - @torch.no_grad() - def get_embeds(self, prompt: List[str]) -> torch.FloatTensor: - input_ids = self.tokenizer( - prompt, - padding="max_length", - max_length=self.tokenizer.model_max_length, - truncation=True, - return_tensors="pt", - ).input_ids - - embeds = self.text_encoder(input_ids)[0] - - return embeds.mean(0)[None] - @torch.no_grad() @replace_example_docstring(EXAMPLE_DOC_STRING) def __call__(