DialoGPT-GPT2-Merge / README.md
johnlorance's picture
Upload folder using huggingface_hub
b60d542 verified
---
license: apache-2.0
tags:
- merge
- mergekit
- lazymergekit
- microsoft/DialoGPT-medium
- gpt2
---
# DialoGPT-GPT2-Merge
DialoGPT-GPT2-Merge is a merge of the following models using [mergekit](https://github.com/cg123/mergekit):
* [microsoft/DialoGPT-medium](https://huggingface.co/microsoft/DialoGPT-medium)
* [gpt2](https://huggingface.co/gpt2)
## 🧩 Configuration
```yaml{'slices': [{'sources': [{'model': 'microsoft/DialoGPT-medium', 'layer_range': [0, 12]}, {'model': 'gpt2', 'layer_range': [0, 12]}]}], 'merge_method': 'slerp', 'base_model': 'microsoft/DialoGPT-medium', 'parameters': {'t': [{'filter': 'self_attn', 'value': [0, 0.5, 0.3, 0.7, 1]}, {'filter': 'mlp', 'value': [1, 0.5, 0.7, 0.3, 0]}, {'value': 0.5}]}, 'dtype': 'bfloat16'}```