Update README.md

README.md

@@ -5,7 +5,7 @@ datasets:
 language:
 - en
 base_model:
-- internlm/internlm2-math-7b
+- internlm/internlm2-math-base-7b
 - deepseek-ai/deepseek-math-7b-base
 pipeline_tag: text-generation
 library_name: transformers
@@ -22,7 +22,7 @@ tags:
     <a href="https://github.com/leanprover-community/mathlib4"><img src="https://img.shields.io/badge/Lean-4-orange" alt="Lean 4"></a>
     <a href="https://github.com/Purewhite2019/rethinking_autoformalization"><img src="https://img.shields.io/badge/GitHub-%23121011.svg?logo=github&logoColor=white" alt="GitHub"></a>
   </div>
-  <h2>Qi Liu, Xinhao Zheng, Xudong Lu, Qinxiang Cao, Junchi Yan* (* indicates Correspondence author)</h2>
+  <h2>Qi Liu, Xinhao Zheng, Xudong Lu, Qinxiang Cao*, Junchi Yan* (* indicates Correspondence author)</h2>
   <h2>Sch. of Computer Science & Sch. of Artificial Intelligence, Shanghai Jiao Tong University</h2>
 </div>
 
@@ -33,7 +33,7 @@ Please refer to the [📺GitHub repo](https://github.com/Purewhite2019/rethinkin
 ## 📈 Performance
 |                       |     **Benchmark**    | **ProofNet** | **Con-NF** |
 |:---------------------:|:--------------------:|--------------|------------|
-| **InternLM2-Math 7B** | Rautoformalizer (-R) | 16.58%       | 4.58%      |
+| **InternLM2-Math-Base 7B** | Rautoformalizer (-R) | 16.58%       | 4.58%      |
 |                       | RAutoformalizer      | 18.18%       | 16.86%     |
 |                       | Rautoformalizer (+R) | 31.28%       | 55.36%     |
 | **DeepseekMath 7B**   | Rautoformalizer (-R) | 15.24%       | 4.27%      |
@@ -43,7 +43,7 @@ Please refer to the [📺GitHub repo](https://github.com/Purewhite2019/rethinkin
 ## ⚙️ Usage
 ### w/o Dependency Retrieval
 - [🤗`purewhite42/rautoformalizer_nora_deepseek`](https://huggingface.co/purewhite42/rautoformalizer_nora_deepseek): w/o dependency retrieval, SFTed from [🤗`deepseek-ai/deepseek-math-7b-base`](https://huggingface.co/deepseek-ai/deepseek-math-7b-base)
-- [🤗`purewhite42/rautoformalizer_nora_internlm`](https://huggingface.co/purewhite42/rautoformalizer_nora_internlm): w/o dependency retrieval, SFTed from [🤗`internlm/internlm2-math-7b`](https://huggingface.co/internlm/internlm2-math-7b)
+- [🤗`purewhite42/rautoformalizer_nora_internlm`](https://huggingface.co/purewhite42/rautoformalizer_nora_internlm): w/o dependency retrieval, SFTed from [🤗`internlm/internlm2-math-base-7b`](https://huggingface.co/internlm/internlm2-math-base-7b)
 
 ```shell
 python -m autoformalizer.autoformalize_vllm_passk \
@@ -62,7 +62,7 @@ python -m autoformalizer.autoformalize_vllm_passk \
 ```
 ### RAutofromalizer (w/ Dependency Retrieval)
 - [🤗`purewhite42/rautoformalizer_ra_deepseek`](https://huggingface.co/purewhite42/rautoformalizer_ra_deepseek): w/ dependency retrieval (`dependency_retriever_f`), SFTed from [🤗`deepseek-ai/deepseek-math-7b-base`](https://huggingface.co/deepseek-ai/deepseek-math-7b-base)
-- [🤗`purewhite42/rautoformalizer_ra_internlm`](https://huggingface.co/purewhite42/rautoformalizer_ra_internlm): w/ dependency retrieval (`dependency_retriever_f`), SFTed from [🤗`internlm/internlm2-math-7b`](https://huggingface.co/internlm/internlm2-math-7b)
+- [🤗`purewhite42/rautoformalizer_ra_internlm`](https://huggingface.co/purewhite42/rautoformalizer_ra_internlm): w/ dependency retrieval (`dependency_retriever_f`), SFTed from [🤗`internlm/internlm2-math-base-7b`](https://huggingface.co/internlm/internlm2-math-base-7b)
 ```shell
 python -m autoformalizer.autoformalize_vllm_w_ra_passk \
     --port ... \    # Can be arbitrarily set (should avoid conflict)
@@ -81,7 +81,7 @@ python -m autoformalizer.autoformalize_vllm_w_ra_passk \
 ```
 ### Oracle RAutoformalizer (w/ Ground-truth Dependencies)
 - [🤗`purewhite42/rautoformalizer_gtra_deepseek`](https://huggingface.co/purewhite42/rautoformalizer_gtra_deepseek): w/ oracle dependency retrieval, SFTed from [🤗`deepseek-ai/deepseek-math-7b-base`](https://huggingface.co/deepseek-ai/deepseek-math-7b-base)
-- [🤗`purewhite42/rautoformalizer_gtra_internlm`](https://huggingface.co/purewhite42/rautoformalizer_gtra_internlm): w/ oracle dependency retrieval, SFTed from [🤗`internlm/internlm2-math-7b`](https://huggingface.co/internlm/internlm2-math-7b)
+- [🤗`purewhite42/rautoformalizer_gtra_internlm`](https://huggingface.co/purewhite42/rautoformalizer_gtra_internlm): w/ oracle dependency retrieval, SFTed from [🤗`internlm/internlm2-math-base-7b`](https://huggingface.co/internlm/internlm2-math-base-7b)
 ```shell
 python -m autoformalizer.autoformalize_vllm_w_gt_passk \
     --port ... \    # Can be arbitrarily set (should avoid conflict)