Skip to content

Commit ea64a78

Browse files
authored
Allow k pipeline to generate > 1 images (#1645)
Allow k pipeline to generate > 1 images.
1 parent 2868d99 commit ea64a78

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)