Skip to content

GitLab

"examples/git@developer.sourcefind.cn:OpenDAS/diffusers.git" did not exist on "c891330f7966fae4aec27962e243a1432d0dc6b5"
Commit 43bf361a authored by patil-suraj's avatar patil-suraj
Browse files

fix more LatentDiffusionPipeline

parent 8199f09c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/pipeline_latent_diffusion.py
View file @ 43bf361a
...@@ -1431,7 +1431,7 @@ class LatentDiffusionPipeline(DiffusionPipeline): ...@@ -1431,7 +1431,7 @@ class LatentDiffusionPipeline(DiffusionPipeline):
uncond_input = self.tokenizer([""], padding="max_length", max_length=77, return_tensors="pt").to( uncond_input = self.tokenizer([""], padding="max_length", max_length=77, return_tensors="pt").to(
torch_device torch_device
) )
uncond_embeddings = self.bert(uncond_input.input_ids)[0] uncond_embeddings = self.bert(uncond_input.input_ids)
# get text embedding # get text embedding
text_input = self.tokenizer(prompt, padding="max_length", max_length=77, return_tensors="pt").to(torch_device) text_input = self.tokenizer(prompt, padding="max_length", max_length=77, return_tensors="pt").to(torch_device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment