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

fix more LatentDiffusionPipeline

Browse Source
This commit is contained in:
patil-suraj
2022-06-27 11:10:10 +02:00
parent 8199f09c22
commit 43bf361a7a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/pipeline_latent_diffusion.py
View File

@@ -1431,7 +1431,7 @@ class LatentDiffusionPipeline(DiffusionPipeline):
uncond_input = self.tokenizer([""], padding="max_length", max_length=77, return_tensors="pt").to(
torch_device
)
uncond_embeddings = self.bert(uncond_input.input_ids)[0]
uncond_embeddings = self.bert(uncond_input.input_ids)
# get text embedding
text_input = self.tokenizer(prompt, padding="max_length", max_length=77, return_tensors="pt").to(torch_device)