add model_stock

app.py

@@ -173,7 +173,16 @@ int8_mask: true
 dtype: bfloat16
 random_seed: 0
 """
-    yaml_config = random.choices([slerp_config, dare_config], weights=[0.4, 0.6], k=1)[0]
+    stock_config = f"""
+models:
+  - model: mistralai/Mistral-7B-v0.1
+  - model: {models[0]["Model"]}
+  - model: {models[1]["Model"]}
+merge_method: model_stock
+base_model: mistralai/Mistral-7B-v0.1
+dtype: bfloat16    
+"""
+    yaml_config = random.choices([slerp_config, dare_config, stock_config], weights=[0.3, 0.3, 0.4], k=1)[0]
 
     with open('config.yaml', 'w', encoding="utf-8") as f:
         f.write(yaml_config)