Commit 7c120874 authored by patil-suraj's avatar patil-suraj
Browse files

fix LatentDiffusionPipeline

parent fbb103de
......@@ -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)
......
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