Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Merge branch 'main' of https://huggingface.co/spaces/AIR-Bench/leaderboard
Browse files- src/about.py +93 -85
src/about.py
CHANGED
|
@@ -30,109 +30,117 @@ pip install .
|
|
| 30 |
```bash
|
| 31 |
cd AIR-Bench/scripts
|
| 32 |
# Run all tasks
|
| 33 |
-
python run_AIR-Bench.py
|
| 34 |
-
--output_dir ./search_results
|
| 35 |
--encoder BAAI/bge-m3 \
|
| 36 |
-
--
|
| 37 |
-
--
|
| 38 |
-
--
|
| 39 |
-
--
|
| 40 |
-
--
|
| 41 |
-
--
|
| 42 |
-
--
|
| 43 |
-
--
|
| 44 |
-
--
|
| 45 |
-
--
|
| 46 |
-
--use_fp16 True \
|
| 47 |
-
--add_instruction False \
|
| 48 |
--overwrite False
|
| 49 |
|
| 50 |
# Run the tasks in the specified task type
|
| 51 |
-
python run_AIR-Bench.py
|
| 52 |
-
--task_types long-doc
|
| 53 |
-
--output_dir ./search_results
|
| 54 |
-
--encoder BAAI/bge-m3
|
| 55 |
-
--
|
| 56 |
-
--
|
| 57 |
-
--
|
| 58 |
-
--
|
| 59 |
-
--
|
| 60 |
-
--
|
| 61 |
-
--
|
| 62 |
-
--
|
| 63 |
-
--
|
| 64 |
-
--
|
| 65 |
-
--use_fp16 True \
|
| 66 |
-
--add_instruction False \
|
| 67 |
--overwrite False
|
| 68 |
|
| 69 |
# Run the tasks in the specified task type and domains
|
| 70 |
-
python run_AIR-Bench.py
|
| 71 |
-
--task_types long-doc
|
| 72 |
-
--domains arxiv book
|
| 73 |
-
--output_dir ./search_results
|
| 74 |
-
--encoder BAAI/bge-m3
|
| 75 |
-
--
|
| 76 |
-
--
|
| 77 |
-
--
|
| 78 |
-
--
|
| 79 |
-
--
|
| 80 |
-
--
|
| 81 |
-
--
|
| 82 |
-
--
|
| 83 |
-
--
|
| 84 |
-
--
|
| 85 |
-
--use_fp16 True \
|
| 86 |
-
--add_instruction False \
|
| 87 |
--overwrite False
|
| 88 |
|
| 89 |
# Run the tasks in the specified languages
|
| 90 |
-
python run_AIR-Bench.py
|
| 91 |
-
--languages en
|
| 92 |
-
--output_dir ./search_results
|
| 93 |
-
--encoder BAAI/bge-m3
|
| 94 |
-
--
|
| 95 |
-
--
|
| 96 |
-
--
|
| 97 |
-
--
|
| 98 |
-
--
|
| 99 |
-
--
|
| 100 |
-
--
|
| 101 |
-
--
|
| 102 |
-
--
|
| 103 |
-
--
|
| 104 |
-
--use_fp16 True \
|
| 105 |
-
--add_instruction False \
|
| 106 |
--overwrite False
|
| 107 |
|
| 108 |
# Run the tasks in the specified task type, domains, and languages
|
| 109 |
-
python run_AIR-Bench.py
|
| 110 |
-
--task_types qa
|
| 111 |
-
--domains wiki web
|
| 112 |
-
--languages en
|
| 113 |
-
--output_dir ./search_results
|
| 114 |
-
--encoder BAAI/bge-m3
|
| 115 |
-
--encoder_link https://huggingface.co/BAAI/bge-m3
|
| 116 |
-
--reranker BAAI/bge-reranker-v2-m3
|
| 117 |
-
--reranker_link https://huggingface.co/BAAI/bge-reranker-v2-m3
|
| 118 |
-
--search_top_k 1000
|
| 119 |
-
--rerank_top_k 100
|
| 120 |
-
--max_query_length 512
|
| 121 |
-
--max_passage_length 512
|
| 122 |
-
--batch_size 512
|
| 123 |
-
--pooling_method cls
|
| 124 |
-
--normalize_embeddings True
|
| 125 |
-
--use_fp16 True
|
| 126 |
-
--add_instruction False
|
| 127 |
--overwrite False
|
| 128 |
```
|
| 129 |
3. Package the search results.
|
| 130 |
```bash
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
--
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 134 |
```
|
| 135 |
-
4. Upload the `.zip` file on this page and fill in the model information
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 136 |
5. Congratulation! Your results will be shown on the leaderboard in up to one hour.
|
| 137 |
"""
|
| 138 |
|
|
|
|
| 30 |
```bash
|
| 31 |
cd AIR-Bench/scripts
|
| 32 |
# Run all tasks
|
| 33 |
+
python run_AIR-Bench.py \\
|
| 34 |
+
--output_dir ./search_results \\
|
| 35 |
--encoder BAAI/bge-m3 \
|
| 36 |
+
--reranker BAAI/bge-reranker-v2-m3 \\
|
| 37 |
+
--search_top_k 1000 \\
|
| 38 |
+
--rerank_top_k 100 \\
|
| 39 |
+
--max_query_length 512 \\
|
| 40 |
+
--max_passage_length 512 \\
|
| 41 |
+
--batch_size 512 \\
|
| 42 |
+
--pooling_method cls \\
|
| 43 |
+
--normalize_embeddings True \\
|
| 44 |
+
--use_fp16 True \\
|
| 45 |
+
--add_instruction False \\
|
|
|
|
|
|
|
| 46 |
--overwrite False
|
| 47 |
|
| 48 |
# Run the tasks in the specified task type
|
| 49 |
+
python run_AIR-Bench.py \\
|
| 50 |
+
--task_types long-doc \\
|
| 51 |
+
--output_dir ./search_results \\
|
| 52 |
+
--encoder BAAI/bge-m3 \\
|
| 53 |
+
--reranker BAAI/bge-reranker-v2-m3 \\
|
| 54 |
+
--search_top_k 1000 \\
|
| 55 |
+
--rerank_top_k 100 \\
|
| 56 |
+
--max_query_length 512 \\
|
| 57 |
+
--max_passage_length 512 \\
|
| 58 |
+
--batch_size 512 \\
|
| 59 |
+
--pooling_method cls \\
|
| 60 |
+
--normalize_embeddings True \\
|
| 61 |
+
--use_fp16 True \\
|
| 62 |
+
--add_instruction False \\
|
|
|
|
|
|
|
| 63 |
--overwrite False
|
| 64 |
|
| 65 |
# Run the tasks in the specified task type and domains
|
| 66 |
+
python run_AIR-Bench.py \\
|
| 67 |
+
--task_types long-doc \\
|
| 68 |
+
--domains arxiv book \\
|
| 69 |
+
--output_dir ./search_results \\
|
| 70 |
+
--encoder BAAI/bge-m3 \\
|
| 71 |
+
--reranker BAAI/bge-reranker-v2-m3 \\
|
| 72 |
+
--search_top_k 1000 \\
|
| 73 |
+
--rerank_top_k 100 \\
|
| 74 |
+
--max_query_length 512 \\
|
| 75 |
+
--max_passage_length 512 \\
|
| 76 |
+
--batch_size 512 \\
|
| 77 |
+
--pooling_method cls \\
|
| 78 |
+
--normalize_embeddings True \\
|
| 79 |
+
--use_fp16 True \\
|
| 80 |
+
--add_instruction False \\
|
|
|
|
|
|
|
| 81 |
--overwrite False
|
| 82 |
|
| 83 |
# Run the tasks in the specified languages
|
| 84 |
+
python run_AIR-Bench.py \\
|
| 85 |
+
--languages en \\
|
| 86 |
+
--output_dir ./search_results \\
|
| 87 |
+
--encoder BAAI/bge-m3 \\
|
| 88 |
+
--reranker BAAI/bge-reranker-v2-m3 \\
|
| 89 |
+
--search_top_k 1000 \\
|
| 90 |
+
--rerank_top_k 100 \\
|
| 91 |
+
--max_query_length 512 \\
|
| 92 |
+
--max_passage_length 512 \\
|
| 93 |
+
--batch_size 512 \\
|
| 94 |
+
--pooling_method cls \\
|
| 95 |
+
--normalize_embeddings True \\
|
| 96 |
+
--use_fp16 True \\
|
| 97 |
+
--add_instruction False \\
|
|
|
|
|
|
|
| 98 |
--overwrite False
|
| 99 |
|
| 100 |
# Run the tasks in the specified task type, domains, and languages
|
| 101 |
+
python run_AIR-Bench.py \\
|
| 102 |
+
--task_types qa \\
|
| 103 |
+
--domains wiki web \\
|
| 104 |
+
--languages en \\
|
| 105 |
+
--output_dir ./search_results \\
|
| 106 |
+
--encoder BAAI/bge-m3 \\
|
| 107 |
+
--encoder_link https://huggingface.co/BAAI/bge-m3 \\
|
| 108 |
+
--reranker BAAI/bge-reranker-v2-m3 \\
|
| 109 |
+
--reranker_link https://huggingface.co/BAAI/bge-reranker-v2-m3 \\
|
| 110 |
+
--search_top_k 1000 \\
|
| 111 |
+
--rerank_top_k 100 \\
|
| 112 |
+
--max_query_length 512 \\
|
| 113 |
+
--max_passage_length 512 \\
|
| 114 |
+
--batch_size 512 \\
|
| 115 |
+
--pooling_method cls \\
|
| 116 |
+
--normalize_embeddings True \\
|
| 117 |
+
--use_fp16 True \\
|
| 118 |
+
--add_instruction False \\
|
| 119 |
--overwrite False
|
| 120 |
```
|
| 121 |
3. Package the search results.
|
| 122 |
```bash
|
| 123 |
+
# Zip "Embedding Model + NoReranker" search results in "<search_results>/<model_name>/NoReranker" to "<save_dir>/<model_name>_NoReranker.zip".
|
| 124 |
+
python zip_results.py \\
|
| 125 |
+
--results_dir search_results \\
|
| 126 |
+
--model_name bge-m3 \\
|
| 127 |
+
--save_dir search_results/zipped_results
|
| 128 |
+
|
| 129 |
+
# Zip "Embedding Model + Reranker" search results in "<search_results>/<model_name>/<reranker_name>" to "<save_dir>/<model_name>_<reranker_name>.zip".
|
| 130 |
+
python zip_results.py \\
|
| 131 |
+
--results_dir search_results \\
|
| 132 |
+
--model_name bge-m3 \\
|
| 133 |
+
--reranker_name bge-reranker-v2-m3 \\
|
| 134 |
+
--save_dir search_results/zipped_results
|
| 135 |
```
|
| 136 |
+
4. Upload the `.zip` file on this page and fill in the model information:
|
| 137 |
+
- Model Name: such as `bge-m3`.
|
| 138 |
+
- Model URL: such as `https://huggingface.co/BAAI/bge-m3`.
|
| 139 |
+
- Reranker Name: such as `bge-reranker-v2-m3`. Keep empty for `NoReranker`.
|
| 140 |
+
- Reranker URL: such as `https://huggingface.co/BAAI/bge-reranker-v2-m3`. Keep empty for `NoReranker`.
|
| 141 |
+
|
| 142 |
+
If you want to stay anonymous, you can only fill in the Model Name and Reranker Name (keep empty for `NoReranker`), and check the selection box below befor submission.
|
| 143 |
+
|
| 144 |
5. Congratulation! Your results will be shown on the leaderboard in up to one hour.
|
| 145 |
"""
|
| 146 |
|