Update splade.py

splade.py

@@ -42,7 +42,7 @@ class Splade(PreTrainedModel):
     similarity = similarity
     encode = encode
 
-    def __init__(self, config, weights_path=None):
+    def __init__(self, config, weights_path=None, token=None):
         super().__init__(config)
         self.name = "splade"
     
@@ -68,6 +68,7 @@ class Splade(PreTrainedModel):
             attn_implementation=config.attn_implementation,
             bidirectional=getattr(config, "bidirectional", False),
             base_cfg=base_cfg,
+            token=token
         )
 
     def save_pretrained(self, save_directory, *args, **kwargs):
@@ -83,7 +84,7 @@ class Splade(PreTrainedModel):
             token=token,
         )
     
-        model = cls(config, weights_path=model_name_or_path)
+        model = cls(config, weights_path=model_name_or_path, token=token)
     
         model.reverse_voc = {v: k for k, v in model.tokenizer.vocab.items()}
         return model