DeepSpeed
103884ae
- Update zero_to_fp32.py (#3936)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
1 year ago
Update zero_to_fp32.py (#3936) Since deepspeed 1 checkpoint structure is identical to deepspeed 2 (AFAIK), we should just change the version check and add support accordingly
References
#3936 - Update zero_to_fp32.py - to support deepspeed_stage_1
Author
PicoCreator
Parents
aef6c65c
Files
1
deepspeed/utils
zero_to_fp32.py
Loading