andstor commited on
Commit
f45a237
1 Parent(s): 1975512

Update src/app.py

Browse files
Files changed (1) hide show
  1. 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