初始化项目,由ModelHub XC社区提供模型
Model: nfaheem/Marcoroni-7b-DPO-Merge Source: Original Platform
This commit is contained in:
100
README.md
Normal file
100
README.md
Normal file
@@ -0,0 +1,100 @@
|
||||
---
|
||||
license: apache-2.0
|
||||
tags:
|
||||
- merge
|
||||
- mergekit
|
||||
- lazymergekit
|
||||
- fblgit/UNA-TheBeagle-7b-v1
|
||||
- udkai/Turdus
|
||||
---
|
||||
|
||||
# Marcoroni-7b-DPO-Merge
|
||||
|
||||
Marcoroni-7b-DPO-Merge is a merge of the following models using [mergekit](https://github.com/cg123/mergekit) and inspired by [Maxime Labonne's work](https://medium.com/@mlabonne):
|
||||
* [fblgit/UNA-TheBeagle-7b-v1](https://huggingface.co/fblgit/UNA-TheBeagle-7b-v1)
|
||||
* [udkai/Turdus](https://huggingface.co/udkai/Turdus)
|
||||
|
||||
## 🧩 Configuration
|
||||
|
||||
```yaml
|
||||
models:
|
||||
- model: madatnlp/marcoroni-7b-v3-safetensor
|
||||
# no parameters necessary for base model
|
||||
- model: fblgit/UNA-TheBeagle-7b-v1
|
||||
parameters:
|
||||
density: 0.3
|
||||
weight: 0.5
|
||||
- model: udkai/Turdus
|
||||
parameters:
|
||||
density: 0.7
|
||||
weight: 0.3
|
||||
merge_method: ties
|
||||
base_model: madatnlp/marcoroni-7b-v3-safetensor
|
||||
parameters:
|
||||
normalize: true
|
||||
dtype: float16
|
||||
```
|
||||
|
||||
|
||||
## 💻 Example Python Code
|
||||
|
||||
```python
|
||||
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
||||
|
||||
model_name_or_path = "nfaheem/Marcoroni-7b-DPO-Merge"
|
||||
model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
|
||||
device_map="auto",
|
||||
revision="main")
|
||||
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
|
||||
|
||||
prompt = "Write a story about llamas"
|
||||
system_message = "You are a story writing assistant"
|
||||
prompt_template=f'''{prompt}
|
||||
'''
|
||||
|
||||
print("\n\n*** Generate:")
|
||||
|
||||
input_ids = tokenizer(prompt_template, return_tensors='pt').input_ids.cuda()
|
||||
output = model.generate(inputs=input_ids, temperature=0.7, do_sample=True, top_p=0.95, top_k=40, max_new_tokens=512)
|
||||
print(tokenizer.decode(output[0]))
|
||||
|
||||
# Inference can also be done using transformers' pipeline
|
||||
|
||||
print("*** Pipeline:")
|
||||
pipe = pipeline(
|
||||
"text-generation",
|
||||
model=model,
|
||||
tokenizer=tokenizer,
|
||||
max_new_tokens=512,
|
||||
do_sample=True,
|
||||
temperature=0.7,
|
||||
top_p=0.95,
|
||||
top_k=40,
|
||||
repetition_penalty=1.1
|
||||
)
|
||||
|
||||
print(pipe(prompt_template)[0]['generated_text'])
|
||||
```
|
||||
|
||||
## 📋 Summary Eval:
|
||||
|
||||
| Average | ARC | HellaSwag | MMLU | TruthfulQA | Winogrande | GSM8K |
|
||||
|---------|-------|-----------|--------|------------|------------|-------|
|
||||
| 74.9 | 73.04 | 88.8 | 64.24 | 70.47 | 85.24 | 67.63 |
|
||||
|
||||
|
||||
## 📈 Huggingface Leaderboard
|
||||
It's Ranked # 1 on HuggingFace Leaderboard among around 13B parameters (01/15/2024)
|
||||
|
||||
| Model | Average | ARC | HellaSwag | MMLU | Truthful QA | Winogrande | GSM8K |
|
||||
| ---------------------------------- | ------- | ----- | --------- | ----- | ----------- | ------------| ----- |
|
||||
| nfaheem/Marcoroni-7b-DPO-Merge | 74.9 | 73.04 | 88.8 | 64.24 | 70.47 | 85.24 | 67.63 |
|
||||
| mlabonne/Beagle14-7b | 74.76 | 72.95 | 87.95 | 64.7 | 68.38 | 82.64 | 71.42 |
|
||||
| udkai/Turdus | 74.66 | 73.38 | 88.56 | 64.52 | 67.11 | 86.66 | 67.7 |
|
||||
| CultriX/MergeTrix-7B | 74.33 | 72.24 | 87.84 | 64.88 | 66.27 | 83.5 | 71.19 |
|
||||
| fblgit/UNA-TheBeagle-7b-v1 | 73.87 | 73.04 | 88 | 63.48 | 69.85 | 82.16 | 66.72 |
|
||||
|
||||
|
||||

|
||||
|
||||
Reference in New Issue
Block a user