Spaces:

Duplicated from hf-accelerate/model-memory-usage

andstor
/

deepspeed-model-memory-usage

Sleeping

App Files Files Community

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

Commit History

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 HF staff commited on Sep 1, 2023