Update app.py
Browse files
app.py
CHANGED
|
@@ -311,7 +311,7 @@ def compute_precise_estimate(
|
|
| 311 |
gpu_choices = list(GPU_PRECISION_PEAKS.keys())
|
| 312 |
precision_choices = ["float32", "float16", "bfloat16", "FP8"]
|
| 313 |
|
| 314 |
-
with gr.Blocks(title="LLM Training Estimator — Chinchilla Scaling Law"
|
| 315 |
gr.Markdown(r"""
|
| 316 |
# 🚀 LLM Training Estimator (Precision-aware)
|
| 317 |
|
|
@@ -424,4 +424,4 @@ with gr.Blocks(title="LLM Training Estimator — Chinchilla Scaling Law", theme=
|
|
| 424 |
""")
|
| 425 |
|
| 426 |
if __name__ == "__main__":
|
| 427 |
-
demo.launch()
|
|
|
|
| 311 |
gpu_choices = list(GPU_PRECISION_PEAKS.keys())
|
| 312 |
precision_choices = ["float32", "float16", "bfloat16", "FP8"]
|
| 313 |
|
| 314 |
+
with gr.Blocks(title="LLM Training Estimator — Chinchilla Scaling Law") as demo:
|
| 315 |
gr.Markdown(r"""
|
| 316 |
# 🚀 LLM Training Estimator (Precision-aware)
|
| 317 |
|
|
|
|
| 424 |
""")
|
| 425 |
|
| 426 |
if __name__ == "__main__":
|
| 427 |
+
demo.launch(theme=gr.themes.Soft(), css=custom_css)
|