Add pipeline tag, library name and link to Github repository (#1)

- Add pipeline tag, library name and link to Github repository (862ceaa2156002b03fa3e37fa966d791ef399813)


Co-authored-by: Niels Rogge <[email protected]>

README.md

@@ -1,15 +1,19 @@
 ---
-license: llama3.2
-datasets:
-- BAAI/Infinity-Instruct
 base_model:
 - meta-llama/Llama-3.2-1B-Instruct
+datasets:
+- BAAI/Infinity-Instruct
+license: llama3.2
+pipeline_tag: text-generation
+library_name: transformers
 ---
 
 ## Model Overview
 
 This weight is a fine-tuned version of **[Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct)** using the **[LLM-Neo](https://arxiv.org/abs/2411.06839)** method. Usage is identical to the original Llama-3.2-1B-Instruct model. 
 
+The official implementation can be found here: https://github.com/yang3121099/LLM-Neo
+
 ## Training Details
 
 The training process employs the **LLM-Neo** method. The dataset is derived from a mixed sample of **[BAAI/Infinity-Instruct](https://huggingface.co/datasets/BAAI/Infinity-Instruct)**, specifically the `0625` and `7M` subsets, with a total of 10k instruction samples. The KD (knowledge distillation) model used is **[Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Llama-3.1-8B-Instruct)**, with the following hyperparameters:
@@ -69,10 +73,7 @@ The mathematical evaluation highlights significant improvements of the current m
 
 ---
 
-
-
 ### Summary
 
 - **Strengths**: The current model demonstrates notable improvements over **Llama-3.2-1B-Instruct** across multiple benchmark tasks, particularly in reasoning and mathematical problem-solving.
-- **Future Directions**: Further optimization in logical reasoning tasks (e.g., **TabMWP**) and continued enhancements in general language and mathematical adaptability.
-
+- **Future Directions**: Further optimization in logical reasoning tasks (e.g., **TabMWP**) and continued enhancements in general language and mathematical adaptability.