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 |
|