models: | |
- model: rirv938/gpt2_sequence_classification_base | |
# no parameters necessary for base model | |
- model: ChaiML/reward_models_100_170000000_cp_332032 | |
parameters: | |
density: 0.9 | |
weight: 0.5 | |
- model: rirv938/reward_gpt2_preference_24m_e2 | |
parameters: | |
density: 0.9 | |
weight: 0.5 | |
merge_method: ties | |
base_model: rirv938/gpt2_sequence_classification_base | |
parameters: | |
normalize: true | |
dtype: float16 | |