models: | |
- model: openai-community/gpt2 | |
parameters: | |
weight: 0.6 | |
- model: microsoft/DialoGPT-small | |
parameters: | |
weight: 0.4 | |
base_model: microsoft/DialoGPT-small | |
merge_method: task_arithmetic | |
dtype: bfloat16 | |
models: | |
- model: openai-community/gpt2 | |
parameters: | |
weight: 0.6 | |
- model: microsoft/DialoGPT-small | |
parameters: | |
weight: 0.4 | |
base_model: microsoft/DialoGPT-small | |
merge_method: task_arithmetic | |
dtype: bfloat16 | |