mlabonne commited on
Commit
a173027
·
verified ·
1 Parent(s): 29f310f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -319,16 +319,16 @@ def merge_loop():
319
 
320
  # Get model name
321
  model_name = get_name(models, USERNAME, version=0)
322
- print("="*20)
323
- print(model_name)
324
 
325
  # Get model license
326
  license = get_license(models)
327
- print(license)
328
 
329
  # Merge configs
330
  yaml_config = create_config(models)
331
- print(yaml_config)
332
 
333
  # Merge models
334
  merge_models()
@@ -360,14 +360,15 @@ sys.stdout = Logger("output.log")
360
  # Gradio interface
361
  title = """
362
  <div align="center">
363
- <p style="font-size: 36px;">♾️ AutoMerger</p>
364
- <p style="font-size: 20px;">📝 <a href="https://medium.com/towards-data-science/merge-large-language-models-with-mergekit-2118fb392b54">Model merging</a> • 💻 <a href="https://github.com/arcee-ai/mergekit">Mergekit</a> • 🐦 <a href="https://twitter.com/maximelabonne">Follow me on X</a></p>
365
  <p><em>AutoMerger selects two 7B models on top of the Open LLM Leaderboard, combine them with a merge technique, and evaluate the resulting model.</em></p>
366
  </div>
367
  """
368
- with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
369
  gr.Markdown(title)
370
- logs = gr.Textbox()
371
  demo.load(read_logs, None, logs, every=1)
372
  demo.launch(server_name="0.0.0.0", prevent_thread_lock=True)
373
 
 
319
 
320
  # Get model name
321
  model_name = get_name(models, USERNAME, version=0)
322
+ print("="*60)
323
+ print(f"Model name: {model_name}")
324
 
325
  # Get model license
326
  license = get_license(models)
327
+ print(f"License: {license}")
328
 
329
  # Merge configs
330
  yaml_config = create_config(models)
331
+ print(f"YAML config:\n{yaml_config}")
332
 
333
  # Merge models
334
  merge_models()
 
360
  # Gradio interface
361
  title = """
362
  <div align="center">
363
+ <p style="font-size: 48px;">♾️ AutoMerger</p>
364
+ <p style="font-size: 20px;">📃 <a href="https://huggingface.co/automerger">Automerged models</a> • 📝 <a href="https://medium.com/towards-data-science/merge-large-language-models-with-mergekit-2118fb392b54">Article</a> • 🐦 <a href="https://twitter.com/maximelabonne">Follow me on X</a></p>
365
  <p><em>AutoMerger selects two 7B models on top of the Open LLM Leaderboard, combine them with a merge technique, and evaluate the resulting model.</em></p>
366
  </div>
367
  """
368
+ css = ".flex-1 {background: #f9fafb;}"
369
+ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
370
  gr.Markdown(title)
371
+ logs = gr.Textbox(label="Logs")
372
  demo.load(read_logs, None, logs, every=1)
373
  demo.launch(server_name="0.0.0.0", prevent_thread_lock=True)
374