mlabonne commited on
Commit
8e43db7
·
verified ·
1 Parent(s): 74224a0

add weight for model_stock

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.3, 0.3, 0.4], k=1)[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)