Update README.md

README.md

@@ -1,8 +1,11 @@
 ---
 license: apache-2.0
+base_model:
+- xai-org/grok-2
 ---
 ### huihui-ai/grok-2
-This Python [script](https://huggingface.co/huihui-ai/grok-2/blob/main/convert_safetensors.py) is designed to process and merge sharded weight files (in safetensors format) for a machine learning model, specifically targeting the `xai-org/grok-2` model. The main functionalities include:
+This Python [script](https://huggingface.co/huihui-ai/grok-2/blob/main/convert_safetensors.py) is designed to process and merge sharded weight files 
+(in safetensors format) for a machine learning model, specifically targeting the `[xai-org/grok-2](https://huggingface.co/xai-org/grok-2)` model. The main functionalities include:
 
 Just a simple merge, without any inference code, and does not indicate whether the final model is reasonable or correct.
 
@@ -35,4 +38,4 @@ Just a simple merge, without any inference code, and does not indicate whether t
 - Ensure the input directory `xai-org/grok-2` contains valid `pytorch_model-*.safetensors` files.
 - The script assumes a tensor parallelism degree of 8 (`tp_count = 8`). Modify the `tp_count` value in the script if needed.
 - Memory requirements may be high; run on a machine with sufficient memory.
-- If shards are missing or shapes mismatch, the script will print warnings and attempt to proceed.
+- If shards are missing or shapes mismatch, the script will print warnings and attempt to proceed.