Update src/app.py
Browse files- src/app.py +1 -1
src/app.py
CHANGED
@@ -37,7 +37,7 @@ def get_results(model_name: str, library: str, precision: list, training: list,
|
|
37 |
with gr.Blocks() as demo:
|
38 |
with gr.Column():
|
39 |
gr.Markdown(
|
40 |
-
"""<img src="https://huggingface.co/spaces/andstor/model-memory-usage/resolve/main/measure_model_size_deepspeed.svg" style="float: left;" width="250" height="250"><h1>🤗 DeepSpeed Model Memory Calculator</h1>
|
41 |
|
42 |
This tool will help you calculate how much memory is required for the various Zero Redundancy Optimizer (ZeRO), given a on a model hosted on the 🤗 Hugging Face Hub and a hardware setup. The optimizer states assume that Adam is used.
|
43 |
|
|
|
37 |
with gr.Blocks() as demo:
|
38 |
with gr.Column():
|
39 |
gr.Markdown(
|
40 |
+
"""<img src="https://huggingface.co/spaces/andstor/deepspeed-model-memory-usage/resolve/main/measure_model_size_deepspeed.svg" style="float: left;" width="250" height="250"><h1>🤗 DeepSpeed Model Memory Calculator</h1>
|
41 |
|
42 |
This tool will help you calculate how much memory is required for the various Zero Redundancy Optimizer (ZeRO), given a on a model hosted on the 🤗 Hugging Face Hub and a hardware setup. The optimizer states assume that Adam is used.
|
43 |
|