rawsh commited on
Commit
d3e76ed
·
verified ·
1 Parent(s): c9446bf

Add padding token to config (fix batched generation)

Browse files

Fixes `Cannot handle batch sizes > 1 if no padding token is defined`

tested with:
```
infinity_emb v2 --model-id michaelfeil/mxbai-rerank-large-v2-seq --device mps --engine torch --no-model-warmup --revision refs/pr/1
```

Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -6,6 +6,7 @@
6
  "attention_dropout": 0.0,
7
  "bos_token_id": 151643,
8
  "eos_token_id": 151645,
 
9
  "hidden_act": "silu",
10
  "hidden_size": 896,
11
  "id2label": {
 
6
  "attention_dropout": 0.0,
7
  "bos_token_id": 151643,
8
  "eos_token_id": 151645,
9
+ "pad_token_id": 151643,
10
  "hidden_act": "silu",
11
  "hidden_size": 896,
12
  "id2label": {