Update README.md

README.md

@@ -1,11 +1,14 @@
 ---
 license: apache-2.0
+base_model:
+- meta-llama/Meta-Llama-3-8B
 ---
-
-To serve the model using vLLM, use the following command:
-```
-import transmla.vllm_registry.deepseek      # from [github repo](https://github.com/MuLabPKU/TransMLA)
-
-from vllm import LLM, SamplingParams
-llm = LLM(model="fxmeng/TransMLA-llama3-8b-8k", trust_remote_code=True)
-```
+
+To serve the model using vLLM, use the following command:
+```
+# from [github repo](https://github.com/MuLabPKU/TransMLA)
+import transmla.vllm_registry.deepseek      
+
+from vllm import LLM, SamplingParams
+llm = LLM(model="fxmeng/TransMLA-llama3-8b-8k", trust_remote_code=True)
+```