Spaces:
Paused
Paused
add weight for model_stock
Browse files
app.py
CHANGED
|
@@ -182,7 +182,7 @@ merge_method: model_stock
|
|
| 182 |
base_model: mistralai/Mistral-7B-v0.1
|
| 183 |
dtype: bfloat16
|
| 184 |
"""
|
| 185 |
-
yaml_config = random.choices([slerp_config, dare_config, stock_config], weights=[0.
|
| 186 |
|
| 187 |
with open('config.yaml', 'w', encoding="utf-8") as f:
|
| 188 |
f.write(yaml_config)
|
|
|
|
| 182 |
base_model: mistralai/Mistral-7B-v0.1
|
| 183 |
dtype: bfloat16
|
| 184 |
"""
|
| 185 |
+
yaml_config = random.choices([slerp_config, dare_config, stock_config], weights=[0.05, 0.05, 0.9], k=1)[0]
|
| 186 |
|
| 187 |
with open('config.yaml', 'w', encoding="utf-8") as f:
|
| 188 |
f.write(yaml_config)
|