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

don't return length

Browse Source
This commit is contained in:
Edna
2025-06-11 20:56:27 -06:00
committed by GitHub
parent 406ab3b1e9
commit 1bd8fdfcb6
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -220,14 +220,12 @@ class ChromaPipeline(
padding="max_length",
max_length=max_sequence_length,
truncation=True,
return_length=True,
return_length=False,
return_overflowing_tokens=False,
return_tensors="pt",
)
text_input_ids = text_inputs.input_ids
untruncated_ids = self.tokenizer(prompt, padding="longest", return_tensors="pt").input_ids
#text_inputs.attention_mask[:, : text_inputs.length + 1] = 1.0
prompt_embeds = self.text_encoder(
text_input_ids.to(device),