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

Browse Source
This commit is contained in:
DN6
2025-06-16 23:38:17 +05:30
parent 188b0d2a2f
commit 602af7411e
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/diffusers/pipelines/chroma/pipeline_chroma.py
View File

@@ -549,6 +549,8 @@ class ChromaPipeline(
self, batch_size, sequence_length, prompt_attention_mask, negative_prompt_attention_mask=None
):
device = prompt_attention_mask.device
# Extend the prompt attention mask to account for image tokens in the final sequence
attention_mask = torch.cat(
[prompt_attention_mask, torch.ones(batch_size, sequence_length, device=device)], dim=1
)
@@ -764,13 +766,6 @@ class ChromaPipeline(
latents,
)
attention_mask, negative_attention_mask = self._prepare_attention_mask(
latents.shape[0],
latents.shape[1],
prompt_attention_mask,
negative_prompt_attention_mask,
)
# 5. Prepare timesteps
sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps) if sigmas is None else sigmas
image_seq_len = latents.shape[1]
@@ -781,6 +776,14 @@ class ChromaPipeline(
self.scheduler.config.get("base_shift", 0.5),
self.scheduler.config.get("max_shift", 1.15),
)
attention_mask, negative_attention_mask = self._prepare_attention_mask(
batch_size=latents.shape[0],
sequence_length=image_seq_len,
prompt_attention_mask=prompt_attention_mask,
negative_prompt_attention_mask=negative_prompt_attention_mask,
)
timesteps, num_inference_steps = retrieve_timesteps(
self.scheduler,
num_inference_steps,