maxoul commited on
Commit
59934d5
·
verified ·
1 Parent(s): 62b3811

Update splade.py

Browse files
Files changed (1) hide show
  1. splade.py +6 -6
splade.py CHANGED
@@ -75,12 +75,12 @@ class Splade(PreTrainedModel):
75
  # local_dir = snapshot_download(model_name_or_path)
76
  # adapter_path = os.path.join(local_dir, "lora")
77
  # model.model.load_adapter(adapter_path)
78
- model.model = PeftModel.from_pretrained(
79
- model.model,
80
- model_name_or_path,
81
- subfolder="lora",
82
- token=kwargs.get("token", None),
83
- )
84
  # model.model = PeftModel.from_pretrained(model.model, adapter_path)
85
  model.reverse_voc = {v: k for k, v in model.tokenizer.vocab.items()}
86
  return model
 
75
  # local_dir = snapshot_download(model_name_or_path)
76
  # adapter_path = os.path.join(local_dir, "lora")
77
  # model.model.load_adapter(adapter_path)
78
+ # model.model = PeftModel.from_pretrained(
79
+ # model.model,
80
+ # model_name_or_path,
81
+ # subfolder="lora",
82
+ # token=kwargs.get("token", None),
83
+ # )
84
  # model.model = PeftModel.from_pretrained(model.model, adapter_path)
85
  model.reverse_voc = {v: k for k, v in model.tokenizer.vocab.items()}
86
  return model