Spaces:
Paused
Paused
Update app.py
Browse files
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("="*
|
| 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:
|
| 364 |
-
<p style="font-size: 20px;"
|
| 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 |
-
|
|
|
|
| 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 |
|