deepspeed-model-memory-usage

Running

andstor commited on Feb 24

Commit

89c1c76

•

1 Parent(s): f020c04

Update src/app.py

Files changed (1) hide show

src/app.py CHANGED Viewed

@@ -39,7 +39,7 @@ with gr.Blocks() as demo:
         gr.Markdown(
             """<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>
-    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.
     To use this tool pass in the URL or model name of the model you want to calculate the memory usage for,
     select which framework it originates from ("auto" will try and detect it from the model metadata), and

         gr.Markdown(
             """<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>
+    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.
     To use this tool pass in the URL or model name of the model you want to calculate the memory usage for,
     select which framework it originates from ("auto" will try and detect it from the model metadata), and