Spaces:

Duplicated from hf-accelerate/model-memory-usage

andstor
/

deepspeed-model-memory-usage

Running

App Files Files Community

deepspeed-model-memory-usage / src /model_utils.py

Commit History

Only do model name translation for Llama-2 and CodeLlama

f8a5ad5
verified

andstor commited on 19 days ago

Update src/model_utils.py

007d5b8
verified

andstor commited on May 19

Update src/model_utils.py

f020c04
verified

andstor commited on Feb 8, 2024

Update calculations

964360b

andstor commited on Feb 5, 2024

Add DeepSpeed ZeRO calculations

674c962

andstor commited on Feb 2, 2024

Refactor code (#11)

0dea527

muellerzr commited on Sep 1, 2023