mlabonne commited on
Commit
380a2f5
1 Parent(s): 2b919a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -180,11 +180,11 @@ random_seed: 0
180
  """
181
  stock_config = f"""
182
  models:
183
- - model: mistralai/Mistral-7B-v0.1
184
  - model: {models[0]["Model"]}
185
  - model: {models[1]["Model"]}
186
  merge_method: model_stock
187
- base_model: mistralai/Mistral-7B-v0.1
188
  dtype: bfloat16
189
  """
190
  yaml_config = random.choices([slerp_config, dare_config, stock_config], weights=[0.3, 0.6, 0.1], k=1)[0]
 
180
  """
181
  stock_config = f"""
182
  models:
183
+ - model: meta-llama/Meta-Llama-3-8B
184
  - model: {models[0]["Model"]}
185
  - model: {models[1]["Model"]}
186
  merge_method: model_stock
187
+ base_model: meta-llama/Meta-Llama-3-8B
188
  dtype: bfloat16
189
  """
190
  yaml_config = random.choices([slerp_config, dare_config, stock_config], weights=[0.3, 0.6, 0.1], k=1)[0]