Spaces:
Running
Running
kovacsvi
commited on
Commit
·
06f6aab
1
Parent(s):
ed88b8b
set max length to 64
Browse files- interfaces/cap.py +1 -1
- 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=
|
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=
|
96 |
)
|
97 |
|
98 |
# JIT trace
|
|
|
92 |
return_tensors="pt",
|
93 |
padding=True,
|
94 |
truncation=True,
|
95 |
+
max_length=64
|
96 |
)
|
97 |
|
98 |
# JIT trace
|