Update splade.py

splade.py

@@ -14,6 +14,7 @@ from .utils import (
     similarity,
     encode,
 )
+from peft import PeftModel
 
 
 class SpladeConfig(PretrainedConfig):
@@ -73,7 +74,8 @@ class Splade(PreTrainedModel):
         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.load_adapter(adapter_path)
+        model.model = PeftModel.from_pretrained(model.model, adapter_path)
         model.reverse_voc = {v: k for k, v in model.tokenizer.vocab.items()}
         return model