kovacsvi commited on
Commit
ed88b8b
·
1 Parent(s): 10307a1

use padding

Browse files
Files changed (1) hide show
  1. interfaces/cap.py +1 -1
interfaces/cap.py CHANGED
@@ -105,7 +105,7 @@ def predict(text, model_id, tokenizer_id):
105
  text,
106
  max_length=256,
107
  truncation=True,
108
- padding="do_not_pad",
109
  return_tensors="pt"
110
  )
111
  inputs = {k: v.to(device) for k, v in inputs.items()}
 
105
  text,
106
  max_length=256,
107
  truncation=True,
108
+ padding=True,
109
  return_tensors="pt"
110
  )
111
  inputs = {k: v.to(device) for k, v in inputs.items()}