Update configure.py

configure.py

@@ -1,11 +1,8 @@
 # configure.py
 
-from transformers import PretrainedConfig
-
 class RecombinationTransformerConfig(PretrainedConfig):
     model_type = "RecombinationTransformer"
-
-    def __init__(self, embed_dim=1024, num_heads=8, num_layers=6, vocab_size=151646, **kwargs):
+    def __init__(self, embed_dim=1024, num_heads=8, num_layers=6, vocab_size=151643, **kwargs):
         super().__init__(**kwargs)
         self.embed_dim = embed_dim
         self.num_heads = num_heads