Update resize padding multiple in test_transformers_logits_vocab_size

dottxt-ai · Oct 3, 2023 · f43a47f · f43a47f
1 parent 1171e68
commit f43a47f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/text/generate/test_integration_transfomers.py b/tests/text/generate/test_integration_transfomers.py
@@ -235,7 +235,7 @@ def test_transformers_logits_vocab_size():
 
     # Artificially increase the weights/logits size relative
     # to the vocabulary
-    model.model.resize_token_embeddings(pad_to_multiple_of=2)
+    model.model.resize_token_embeddings(pad_to_multiple_of=3)
 
     assert len(model.tokenizer.vocabulary) == 1024
     assert model.model.base_model.wte.weight.shape[0] == 1026