Update app.py

app.py

@@ -147,7 +147,7 @@ slices:
       - model: {models[1]["Model"]}
         layer_range: [0, 32]
 merge_method: slerp
-base_model: {models[0]["Model"]}
+base_model: mlabonne/Meta-Llama-3-8B
 parameters:
   t:
     - filter: self_attn
@@ -160,14 +160,18 @@ random_seed: 0
     """
     dare_config = f"""
 models:
-- model: {models[0]["Model"]}
+- model: mlabonne/Meta-Llama-3-8B
   # No parameters necessary for base model
+- model: {models[0]["Model"]}
+  parameters:
+    density: 0.53
+    weight: 0.5
 - model: {models[1]["Model"]}
   parameters:
     density: 0.53
-    weight: 0.6
+    weight: 0.5
 merge_method: dare_ties
-base_model: {models[0]["Model"]}
+base_model: mlabonne/Meta-Llama-3-8B
 parameters:
 int8_mask: true
 dtype: bfloat16
@@ -175,14 +179,14 @@ random_seed: 0
 """ 
     stock_config = f"""
 models:
-  - model: mistralai/Mistral-7B-v0.1
+  - model: mlabonne/Meta-Llama-3-8B
   - model: {models[0]["Model"]}
   - model: {models[1]["Model"]}
 merge_method: model_stock
-base_model: mistralai/Mistral-7B-v0.1
+base_model: mlabonne/Meta-Llama-3-8B
 dtype: bfloat16    
 """
-    yaml_config = random.choices([slerp_config, dare_config, stock_config], weights=[0.01, 0.01, 0.98], k=1)[0]
+    yaml_config = random.choices([slerp_config, dare_config, stock_config], weights=[0.5, 0.4, 0.1], k=1)[0]
 
     with open('config.yaml', 'w', encoding="utf-8") as f:
         f.write(yaml_config)