AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Update src/diffusers/pipelines/kandinsky5/pipeline_kandinsky.py

Browse Source 
Co-authored-by: YiYi Xu <yixu310@gmail.com>
This commit is contained in:
Lev Novitskiy
2025-10-16 09:54:00 +03:00
committed by GitHub
parent 588c12ab98
commit 9b06afba6b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/kandinsky5/pipeline_kandinsky.py
View File

@@ -314,7 +314,7 @@ class Kandinsky5T2VPipeline(DiffusionPipeline, KandinskyLoraLoaderMixin):
attention_mask = inputs["attention_mask"][:, crop_start:]
cu_seqlens = torch.cumsum(attention_mask.sum(1), dim=0)
cu_seqlens = torch.cat([torch.zeros_like(cu_seqlens)[:1], cu_seqlens]).to(dtype=torch.int32)
embeds = torch.cat([embeds[i].unsqueeze(dim=0).repeat(num_videos_per_prompt, 1, 1) for i in range(batch_size)], dim=0)
embeds = embeds.repeat_interleave(num_videos_per_prompt, dim=0)
return embeds.to(dtype), cu_seqlens