Update app.py

app.py

@@ -173,16 +173,7 @@ 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, stock_config], weights=[0.05, 0.05, 0.9], k=1)[0]
+    yaml_config = random.choices([slerp_config, dare_config], weights=[0.4, 0.6], k=1)[0]
 
     with open('config.yaml', 'w', encoding="utf-8") as f:
         f.write(yaml_config)