Tom Aarsen
commited on
Commit
•
c0c6d64
1
Parent(s):
9862f98
Use self.max_seq_length to inform the maximum tokenize length
Browse files- custom_st.py +1 -1
custom_st.py
CHANGED
|
@@ -77,7 +77,7 @@ class MultiModalTransformer(BaseTransformer):
|
|
| 77 |
all_texts,
|
| 78 |
padding="longest",
|
| 79 |
truncation=True,
|
| 80 |
-
max_length=
|
| 81 |
return_tensors="pt",
|
| 82 |
)
|
| 83 |
if all_images:
|
|
|
|
| 77 |
all_texts,
|
| 78 |
padding="longest",
|
| 79 |
truncation=True,
|
| 80 |
+
max_length=self.max_seq_length,
|
| 81 |
return_tensors="pt",
|
| 82 |
)
|
| 83 |
if all_images:
|