kovacsvi commited on
Commit
06f6aab
·
1 Parent(s): ed88b8b

set max length to 64

Browse files
Files changed (2) hide show
  1. interfaces/cap.py +1 -1
  2. utils.py +1 -1
interfaces/cap.py CHANGED
@@ -103,7 +103,7 @@ def predict(text, model_id, tokenizer_id):
103
  t2 = time.perf_counter()
104
  inputs = tokenizer(
105
  text,
106
- max_length=256,
107
  truncation=True,
108
  padding=True,
109
  return_tensors="pt"
 
103
  t2 = time.perf_counter()
104
  inputs = tokenizer(
105
  text,
106
+ max_length=64,
107
  truncation=True,
108
  padding=True,
109
  return_tensors="pt"
utils.py CHANGED
@@ -92,7 +92,7 @@ def download_hf_models():
92
  return_tensors="pt",
93
  padding=True,
94
  truncation=True,
95
- max_length=256
96
  )
97
 
98
  # JIT trace
 
92
  return_tensors="pt",
93
  padding=True,
94
  truncation=True,
95
+ max_length=64
96
  )
97
 
98
  # JIT trace