Correct pipeline tag and add library name
Browse filesThis PR corrects the `pipeline_tag` to `feature-extraction` which is more appropriate for a reranking model. It also adds the `library_name` as `transformers` given the provided usage examples.
README.md
CHANGED
@@ -1,15 +1,16 @@
|
|
1 |
---
|
2 |
-
license: mit
|
3 |
-
datasets:
|
4 |
-
- jhu-clsp/rank1-training-data
|
5 |
base_model:
|
6 |
- Qwen/Qwen2.5-14B
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
8 |
tags:
|
9 |
- reranker
|
10 |
- retrieval
|
11 |
-
|
12 |
-
- en
|
13 |
---
|
14 |
|
15 |
# rank1-14b: Test-Time Compute for Reranking in Information Retrieval
|
@@ -177,4 +178,4 @@ If you use rank1 in your research, please cite our work:
|
|
177 |
|
178 |
## License
|
179 |
|
180 |
-
[MIT License](https://github.com/orionw/rank1/blob/main/LICENSE)
|
|
|
1 |
---
|
|
|
|
|
|
|
2 |
base_model:
|
3 |
- Qwen/Qwen2.5-14B
|
4 |
+
datasets:
|
5 |
+
- jhu-clsp/rank1-training-data
|
6 |
+
language:
|
7 |
+
- en
|
8 |
+
license: mit
|
9 |
+
pipeline_tag: feature-extraction
|
10 |
tags:
|
11 |
- reranker
|
12 |
- retrieval
|
13 |
+
library_name: transformers
|
|
|
14 |
---
|
15 |
|
16 |
# rank1-14b: Test-Time Compute for Reranking in Information Retrieval
|
|
|
178 |
|
179 |
## License
|
180 |
|
181 |
+
[MIT License](https://github.com/orionw/rank1/blob/main/LICENSE)
|