Update splade.py
Browse files
splade.py
CHANGED
|
@@ -22,10 +22,10 @@ class SpladeConfig(PretrainedConfig):
|
|
| 22 |
|
| 23 |
def __init__(
|
| 24 |
self,
|
| 25 |
-
model_name_or_path: str = "
|
| 26 |
attn_implementation: str = "flash_attention_2",
|
| 27 |
bidirectional: bool = True, # only for decoder models
|
| 28 |
-
padding_side: str = "
|
| 29 |
**kwargs,
|
| 30 |
):
|
| 31 |
super().__init__(**kwargs)
|
|
|
|
| 22 |
|
| 23 |
def __init__(
|
| 24 |
self,
|
| 25 |
+
model_name_or_path: str = "Qwen/Qwen3-0.6B",
|
| 26 |
attn_implementation: str = "flash_attention_2",
|
| 27 |
bidirectional: bool = True, # only for decoder models
|
| 28 |
+
padding_side: str = "left",
|
| 29 |
**kwargs,
|
| 30 |
):
|
| 31 |
super().__init__(**kwargs)
|