Skip to content

Commit de384ff

Browse files
committed
Remove warning about pad_token_id for moondream
1 parent 0031705 commit de384ff

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

taggui/widgets/auto_captioner.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -525,8 +525,9 @@ def get_moondream_inputs(model, processor, text: str, pil_image: PilImage,
525525
'inputs_embeds': inputs_embeds,
526526
'attention_mask': (torch.ones(1, inputs_embeds.shape[1]).bool()
527527
.to(device, **dtype_argument)),
528+
'bos_token_id': processor.bos_token_id,
528529
'eos_token_id': eos_tokens_ids,
529-
'bos_token_id': processor.bos_token_id
530+
'pad_token_id': eos_tokens_ids[0]
530531
}
531532
return model_inputs
532533

0 commit comments

Comments
 (0)