Update splade.py

splade.py

@@ -70,18 +70,15 @@ class Splade(PreTrainedModel):
 
     @classmethod
     def from_pretrained(cls, model_name_or_path, *args, **kwargs):
-        config = SpladeConfig.from_pretrained(model_name_or_path)
-        model = cls(config)
-        # local_dir = snapshot_download(model_name_or_path)
-        # adapter_path = os.path.join(local_dir, "lora")
-        # model.model.load_adapter(adapter_path)
-        # model.model = PeftModel.from_pretrained(
-        #     model.model,
-        #     model_name_or_path,
-        #     subfolder="lora",
-        #     token=kwargs.get("token", None),
-        # )
-        # model.model = PeftModel.from_pretrained(model.model, adapter_path)
+        token = kwargs.get("token", None)
+    
+        config = SpladeConfig.from_pretrained(
+            model_name_or_path,
+            token=token,
+        )
+    
+        model = cls(config, weights_path=model_name_or_path)
+    
         model.reverse_voc = {v: k for k, v in model.tokenizer.vocab.items()}
         return model