DeepSpeed
556f0051 - Fix random token-generation issue + MP-checkpoint loading/saving (#2132)

Comment changes are shownComment changes are hidden
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>
Parents
  • deepspeed
    • File
      __init__.py
    • inference
      • File
        engine.py
    • module_inject
      • File
        load_checkpoint.py
      • File
        replace_module.py
      • File
        replace_policy.py
    • ops/transformer/inference
      • File
        transformer_inference.py
    • runtime
      • File
        state_dict_factory.py