Skip to content

Commit ac4b099

Browse files
Apply style fixes
1 parent 1fe64c2 commit ac4b099

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

src/diffusers/pipelines/chroma/pipeline_chroma_img2img.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -260,9 +260,7 @@ def _get_t5_prompt_embeds(
260260
prompt_embeds = prompt_embeds.to(dtype=dtype, device=device)
261261

262262
seq_lengths = tokenizer_mask_device.sum(dim=1)
263-
mask_indices = torch.arange(tokenizer_mask_device.size(1), device=device).unsqueeze(0).expand(
264-
batch_size, -1
265-
)
263+
mask_indices = torch.arange(tokenizer_mask_device.size(1), device=device).unsqueeze(0).expand(batch_size, -1)
266264
attention_mask = (mask_indices <= seq_lengths.unsqueeze(1)).to(dtype=dtype, device=device)
267265

268266
_, seq_len, _ = prompt_embeds.shape

0 commit comments

Comments
 (0)