Update splade.py
Browse files
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(
|