Hugging Face's logo

Spaces:
andstor
/
deepspeed-model-memory-usage
Running

App Files Community
andstor commited on
Commit
65ae2f6
·
verified ·
1 Parent(s): 007d5b8

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +1 -1
src/app.py CHANGED
@@ -36,7 +36,7 @@ def get_results(model_name: str, library: str, precision: list, training: list,
36
 
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 model hosted on the 🤗 Hugging Face Hub and a hardware setup. The optimizer states assume that Adam is used.
 
36
 
37
  with gr.Blocks() as demo:
38
  with gr.Column():
39
+ gr.HTML(
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 model hosted on the 🤗 Hugging Face Hub and a hardware setup. The optimizer states assume that Adam is used.