DeepSpeed
556f0051
- Fix random token-generation issue + MP-checkpoint loading/saving (#2132)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
2 years ago
Fix random token-generation issue + MP-checkpoint loading/saving (#2132) Co-authored-by: Jeff Rasley <jerasley@microsoft.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
References
#2132 - Fix random token-generation issue + MP-checkpoint loading/saving
Author
RezaYazdaniAminabadi
Parents
57140e8e
Files
7
deepspeed
__init__.py
inference
engine.py
module_inject
load_checkpoint.py
replace_module.py
replace_policy.py
ops/transformer/inference
transformer_inference.py
runtime
state_dict_factory.py
Loading