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)
|