Update app.py

app.py

@@ -172,8 +172,17 @@ parameters:
 int8_mask: true
 dtype: bfloat16
 random_seed: 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], weights=[0.4, 0.6], k=1)[0]
+    yaml_config = random.choices([slerp_config, dare_config, stock_config], weights=[0.01, 0.01, 0.98], k=1)[0]
 
     with open('config.yaml', 'w', encoding="utf-8") as f:
         f.write(yaml_config)