Skip to content

Commit f68c954

Browse files
pcuencasliard
authored andcommitted
Allow k pipeline to generate > 1 images (huggingface#1645)
Allow k pipeline to generate > 1 images.
1 parent 059cf71 commit f68c954

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -439,6 +439,7 @@ def __call__(
439439
# 6. Define model function
440440
def model_fn(x, t):
441441
latent_model_input = torch.cat([x] * 2)
442+
t = torch.cat([t] * 2)
442443

443444
noise_pred = self.k_diffusion_model(latent_model_input, t, cond=text_embeddings)
444445

0 commit comments

Comments
 (0)