Update splade.py

splade.py

@@ -72,10 +72,16 @@ class Splade(PreTrainedModel):
     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")
+        # 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, 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)
         model.reverse_voc = {v: k for k, v in model.tokenizer.vocab.items()}
         return model