Fix tokenizer test and make model tests smaller

huggingface · Aug 5, 2021 · 3fa1768 · 3fa1768
1 parent 3412065
commit 3fa1768
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/transformers/models/layoutlmv2/tokenization_layoutlmv2.py b/src/transformers/models/layoutlmv2/tokenization_layoutlmv2.py
@@ -193,7 +193,12 @@ def __init__(
             pad_token=pad_token,
             cls_token=cls_token,
             mask_token=mask_token,
+            cls_token_box=cls_token_box,
+            sep_token_box=sep_token_box,
+            pad_token_box=pad_token_box,
+            pad_token_label=pad_token_label,
             tokenize_chinese_chars=tokenize_chinese_chars,
+            strip_accents=strip_accents,
             model_max_length=model_max_length,
             additional_special_tokens=additional_special_tokens,
             **kwargs,

diff --git a/tests/test_modeling_layoutlmv2.py b/tests/test_modeling_layoutlmv2.py
@@ -44,17 +44,17 @@ class LayoutLMv2ModelTester:
     def __init__(
         self,
         parent,
-        batch_size=13,
+        batch_size=2,
         num_channels=3,
-        image_size=32,
+        image_size=4,
         seq_length=7,
         is_training=True,
         use_input_mask=True,
         use_token_type_ids=True,
         use_labels=True,
         vocab_size=99,
         hidden_size=36,
-        num_hidden_layers=5,
+        num_hidden_layers=3,
         num_attention_heads=4,
         intermediate_size=37,
         hidden_act="gelu",