From 7c120874bedb11fc6fda619e99dc8ab6fb922f48 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Mon, 27 Jun 2022 11:09:21 +0200 Subject: [PATCH 1/2] fix LatentDiffusionPipeline --- src/diffusers/pipelines/pipeline_latent_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pipeline_latent_diffusion.py b/src/diffusers/pipelines/pipeline_latent_diffusion.py index 219ad31d04..aa0ec3810e 100644 --- a/src/diffusers/pipelines/pipeline_latent_diffusion.py +++ b/src/diffusers/pipelines/pipeline_latent_diffusion.py @@ -1435,7 +1435,7 @@ class LatentDiffusionPipeline(DiffusionPipeline): # get text embedding text_input = self.tokenizer(prompt, padding="max_length", max_length=77, return_tensors="pt").to(torch_device) - text_embedding = self.bert(text_input.input_ids)[0] + text_embedding = self.bert(text_input.input_ids) num_trained_timesteps = self.noise_scheduler.config.timesteps inference_step_times = range(0, num_trained_timesteps, num_trained_timesteps // num_inference_steps) From 43bf361a7a9e652ed2982198652145a3d67acb31 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Mon, 27 Jun 2022 11:10:10 +0200 Subject: [PATCH 2/2] fix more LatentDiffusionPipeline --- src/diffusers/pipelines/pipeline_latent_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pipeline_latent_diffusion.py b/src/diffusers/pipelines/pipeline_latent_diffusion.py index aa0ec3810e..ffc8ae670c 100644 --- a/src/diffusers/pipelines/pipeline_latent_diffusion.py +++ b/src/diffusers/pipelines/pipeline_latent_diffusion.py @@ -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)