Update README.md

README.md

@@ -1,88 +1,111 @@
 ---
 tags:
-- text-generation
+    - text-generation
 license: cc-by-nc-sa-4.0
 language:
-- ko
+    - ko
 base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
 pipeline_tag: text-generation
-widget:
-  - text: >
-      <|system|>
-
-      You are a chatbot who answers User's questions.</s>
-
-      <|user|>
-
-      대한민국의 수도는 어디야?</s>
-
-      <|assistant|>
+datasets:
+    - beomi/KoAlpaca-v1.1a
+    - jojo0217/korean_rlhf_dataset
+    - kyujinpy/OpenOrca-KO
+    - nlpai-lab/kullm-v2
 ---
 
 # **DataVortexTL-1.1B-v0.1**
-<img src="./DataVortex.png" alt="DataVortex" style="height: 8em;">
 
-## **License**
-
-[cc-by-nc-sa-4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/)
+<img src="./DataVortex.png" alt="DataVortex" style="height: 8em;">
 
 ## **Model Details**
 
 ### **Base Model**
-[TinyLlama/TinyLlama-1.1B-Chat-v1.0](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v1.0)  
+
+[TinyLlama/TinyLlama-1.1B-Chat-v1.0](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v1.0)
 
 ### **Trained On**
-H100 80GB 1ea
+
+-   **OS**: Ubuntu 20.04
+-   **GPU**: H100 80GB 1ea
+-   **transformers**: v4.36.2
+
+### **Dataset**
+
+-   [beomi/KoAlpaca-v1.1a](https://huggingface.co/datasets/beomi/KoAlpaca-v1.1a)
+-   [jojo0217/korean_rlhf_dataset](https://huggingface.co/datasets/jojo0217/korean_rlhf_dataset)
+-   [kyujinpy/OpenOrca-KO](https://huggingface.co/datasets/kyujinpy/OpenOrca-KO)
+-   [nlpai-lab/kullm-v2](https://huggingface.co/datasets/nlpai-lab/kullm-v2)
 
 ### **Instruction format**
 
-<!-- It follows **(No Input) Alpaca** format. -->
+It follows **TinyLlama** format.
+
+E.g.
+
+```python
+text = """\
+<|system|>
+당신은 사람들이 정보를 찾을 수 있도록 도와주는 인공지능 비서입니다.</s>
+<|user|>
+대한민국의 수도는 어디야?</s>
+<|assistant|>
+대한민국의 수도는 서울입니다.</s>
+<|user|>
+서울 인구는 총 몇 명이야?</s>
+"""
+```
 
 ## **Model Benchmark**
 
-### **Ko-LLM-Leaderboard**
+### **[Ko-LLM-Leaderboard](https://huggingface.co/spaces/upstage/open-ko-llm-leaderboard)**
+
+On Benchmarking ...
 
-On Benchmarking...
+| Model                        | Average | Ko-ARC | Ko-HellaSwag | Ko-MMLU | Ko-TruthfulQA | Ko-CommonGen V2 |
+| ---------------------------- | ------- | ------ | ------------ | ------- | ------------- | --------------- |
+| DataVortexM-7B-Instruct-v0.1 | 39.81   | 34.13  | 42.35        | 38.73   | 45.46         | 38.37           |
+| DataVortexS-10.7B-v0.1       | 0       | 0      | 0            | 0       | 0             | 0               |
+| DataVortexS-10.7B-v0.2       | 43.6    | 38.74  | 50.74        | 38.98   | 44.7          | 44.86           |
+| DataVortexS-10.7B-v0.3       | 0       | 0      | 0            | 0       | 0             | 0               |
+| DataVortexS-10.7B-v0.4       | 0       | 0      | 0            | 0       | 0             | 0               |
+| DataVortexS-10.7B-v0.4       | 0       | 0      | 0            | 0       | 0             | 0               |
+| **DataVortexTL-1.1B-v0.1**   | **0**   | **0**  | **0**        | **0**   | **0**         | **0**           |
+| DataVortexS-10.7B-dpo-v0.1   | 0       | 0      | 0            | 0       | 0             | 0               |
 
-# **Implementation Code**
+## **Implementation Code**
 
-Since, chat_template already contains insturction format above.
+This model contains the chat_template instruction format.  
 You can use the code below.
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-device = "cuda"
+device = "cuda" # the device to load the model onto
 
-model = AutoModelForCausalLM.from_pretrained("Edentns/DataVortexTL-1.1B-v0.1", device_map=device)
+model = AutoModelForCausalLM.from_pretrained("Edentns/DataVortexTL-1.1B-v0.1")
 tokenizer = AutoTokenizer.from_pretrained("Edentns/DataVortexTL-1.1B-v0.1")
 
 messages = [
-    { "role": "user", "content": "대한민국의 수도는 어디야?" }
+    {"role": "system", "content": "당신은 사람들이 정보를 찾을 수 있도록 도와주는 인공지능 비서입니다."},
+    {"role": "user", "content": "대한민국의 수도는 어디야?"},
+    {"role": "assistant", "content": "대한민국의 수도는 서울입니다."},
+    {"role": "user", "content": "서울 인구는 총 몇 명이야?"}
 ]
 
-encoded = tokenizer.apply_chat_template(
-    messages,
-    add_generation_prompt=True,
-    return_tensors="pt",
-    return_token_type_ids=False
-).to(device)
-
-decoded = model.generate(
-    input_ids=encoded,
-    temperature=0.2,
-    top_p=0.9,
-    repetition_penalty=1.2,
-    do_sample=True,
-    max_length=4096,
-    eos_token_id=tokenizer.eos_token_id,
-    pad_token_id=tokenizer.eos_token_id
-)
-decoded = decoded[0][encoded.shape[1]:decoded[0].shape[-1]]
-decoded_text = tokenizer.decode(decoded, skip_special_tokens=True)
-print(decoded_text)
+encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt")
+
+model_inputs = encodeds.to(device)
+model.to(device)
+
+generated_ids = model.generate(model_inputs, max_new_tokens=1000, do_sample=True)
+decoded = tokenizer.batch_decode(generated_ids)
+print(decoded[0])
 ```
 
+## **License**
+
+The model is licensed under the [cc-by-nc-sa-4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/) license, which allows others to copy, modify, and share the work non-commercially, as long as they give appropriate credit and distribute any derivative works under the same license.
+
 <div align="center">
     <a href="https://edentns.com/">
         <img src="./Logo.png" alt="Logo" style="height: 3em;">