Upload 3 files
Browse files
configuration_minicpm_reranker.py
CHANGED
|
@@ -172,7 +172,7 @@ class MiniCPMConfig(PretrainedConfig):
|
|
| 172 |
|
| 173 |
self.start_layer = start_layer
|
| 174 |
self.head_multi = head_multi
|
| 175 |
-
self.
|
| 176 |
|
| 177 |
super().__init__(
|
| 178 |
pad_token_id=pad_token_id,
|
|
|
|
| 172 |
|
| 173 |
self.start_layer = start_layer
|
| 174 |
self.head_multi = head_multi
|
| 175 |
+
self.head_type = head_type
|
| 176 |
|
| 177 |
super().__init__(
|
| 178 |
pad_token_id=pad_token_id,
|