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

update to pad tokens

Browse Source
This commit is contained in:
DN6
2025-06-13 13:43:17 +05:30
parent 74fe45e823
commit 926dcc6319
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -226,7 +226,14 @@ class ChromaPipeline(
return_overflowing_tokens=False,
return_tensors="pt",
)
text_input_ids = text_inputs.input_ids + self.tokenizer.pad_token_id
pad_token_id = self.tokenizer.pad_token_id
text_input_ids = torch.cat(
[
text_inputs.input_ids,
torch.full((text_inputs.input_ids.size(0), 1), pad_token_id, dtype=text_inputs.input_ids.dtype),
],
dim=1,
)
prompt_embeds = self.text_encoder(
text_input_ids.to(device),