AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Revert "[Pix2Pix0] Add utility function to get edit vector" (#2384)

Browse Source 
Revert "[Pix2Pix0] Add utility function to get edit vector (#2383)"

This reverts commit 857c04cfba.
This commit is contained in:
Patrick von Platen
2023-02-17 00:00:27 +02:00
committed by GitHub
parent 857c04cfba
commit 8e1cae5d66
1 changed files with 0 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py
View File

@@ -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__(