maxoul commited on
Commit
03203c6
·
verified ·
1 Parent(s): 3a87604

Update splade.py

Browse files
Files changed (1) hide show
  1. splade.py +1 -1
splade.py CHANGED
@@ -42,7 +42,7 @@ class Splade(PreTrainedModel):
42
  similarity = similarity
43
  encode = encode
44
 
45
- def __init__(self, config):
46
  super().__init__(config)
47
  self.name = "splade"
48
  base_cfg = AutoConfig.from_pretrained(
 
42
  similarity = similarity
43
  encode = encode
44
 
45
+ def __init__(self, config, weights_path=None):
46
  super().__init__(config)
47
  self.name = "splade"
48
  base_cfg = AutoConfig.from_pretrained(