|
--- |
|
tags: |
|
- text-generation |
|
license: cc-by-nc-4.0 |
|
language: |
|
- ko |
|
base_model: upstage/SOLAR-10.7B-Instruct-v1.0 |
|
pipeline_tag: text-generation |
|
--- |
|
|
|
# **DataVortexS-10.7B-dpo-v1.1** |
|
|
|
<img src="./DataVortex.png" alt="DataVortex" style="height: 8em;"> |
|
|
|
## **Model Details** |
|
|
|
### **Base Model** |
|
|
|
[upstage/SOLAR-10.7B-Instruct-v1.0](https://huggingface.co/upstage/SOLAR-10.7B-Instruct-v1.0) |
|
|
|
### **Trained On** |
|
|
|
- **OS**: Ubuntu 22.04 |
|
- **GPU**: H100 80GB 4ea |
|
- **transformers**: v4.36.2 |
|
|
|
### **Instruction format** |
|
|
|
It follows **Alpaca (Chat)** format. |
|
|
|
E.g. |
|
|
|
```python |
|
text = """\ |
|
### System: |
|
λΉμ μ μ¬λλ€μ΄ μ 보λ₯Ό μ°Ύμ μ μλλ‘ λμμ£Όλ μΈκ³΅μ§λ₯ λΉμμ
λλ€. |
|
|
|
### User: |
|
λνλ―Όκ΅μ μλλ μ΄λμΌ? |
|
|
|
### Assistant: |
|
λνλ―Όκ΅μ μλλ μμΈμ
λλ€. |
|
|
|
### User: |
|
μμΈ μΈκ΅¬λ μ΄ λͺ λͺ
μ΄μΌ? |
|
""" |
|
``` |
|
|
|
## **Model Benchmark** |
|
|
|
### **[Ko LM Eval Harness](https://github.com/Beomi/ko-lm-evaluation-harness)** |
|
|
|
| Task | 0-shot | 5-shot | 10-shot | 50-shot | |
|
| :--------------- | -----------: | -------------: | ------------: | -----------: | |
|
| kobest_boolq | 0.375807 | 0.822623 | 0.828582 | 0.822529 | |
|
| kobest_copa | 0.539993 | 0.665979 | 0.67998 | 0.694997 | |
|
| kobest_hellaswag | 0.405785 | 0.401975 | 0.438219 | 0.402962 | |
|
| kobest_sentineg | 0.794083 | 0.85276 | 0.883509 | 0.880932 | |
|
| **Average** | **0.528917** | **0.68583425** | **0.7075725** | **0.700355** | |
|
|
|
### **[Ko-LLM-Leaderboard](https://huggingface.co/spaces/upstage/open-ko-llm-leaderboard)** |
|
|
|
On Benchmarking ... |
|
|
|
| Average | Ko-ARC | Ko-HellaSwag | Ko-MMLU | Ko-TruthfulQA | Ko-CommonGen V2 | |
|
| ------: | -----: | -----------: | ------: | ------------: | --------------: | |
|
| 0 | 0 | 0 | 0 | 0 | 0 | |
|
|
|
## **Implementation Code** |
|
|
|
This model contains the chat_template instruction format. |
|
You can use the code below. |
|
|
|
```python |
|
from transformers import AutoModelForCausalLM, AutoTokenizer |
|
|
|
device = "cuda" # the device to load the model onto |
|
|
|
model = AutoModelForCausalLM.from_pretrained("Edentns/DataVortexS-10.7B-dpo-v1.1") |
|
tokenizer = AutoTokenizer.from_pretrained("Edentns/DataVortexS-10.7B-dpo-v1.1") |
|
|
|
messages = [ |
|
{"role": "system", "content": "λΉμ μ μ¬λλ€μ΄ μ 보λ₯Ό μ°Ύμ μ μλλ‘ λμμ£Όλ μΈκ³΅μ§λ₯ λΉμμ
λλ€."}, |
|
{"role": "user", "content": "λνλ―Όκ΅μ μλλ μ΄λμΌ?"}, |
|
{"role": "assistant", "content": "λνλ―Όκ΅μ μλλ μμΈμ
λλ€."}, |
|
{"role": "user", "content": "μμΈ μΈκ΅¬λ μ΄ λͺ λͺ
μ΄μΌ?"} |
|
] |
|
|
|
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** |
|
|
|
This model is licensed under the [upstage/SOLAR-10.7B-Instruct-v1.0](https://huggingface.co/upstage/SOLAR-10.7B-Instruct-v1.0) license, with the [cc-by-nc-4.0](https://creativecommons.org/licenses/by-nc/4.0/) license granted. Under this license, others are allowed to copy, modify, and share the work, as long as it is not used for commercial purposes. They must provide appropriate credit and distribute any derivative works under the same license. For more details, please refer to the [cc-by-nc-4.0](https://creativecommons.org/licenses/by-nc/4.0/) license. |
|
|
|
<div align="center"> |
|
<a href="https://edentns.com/"> |
|
<img src="./Logo.png" alt="Logo" style="height: 3em;"> |
|
</a> |
|
</div> |
|
|