DeepSpeed
0bafeac4 - enable autoTP for MPT (#3861)

Commit
2 years ago
enable autoTP for MPT (#3861) * enable autoTP for MPT Signed-off-by: Wang, Yi A <yi.a.wang@intel.com> * add model specific func to auto_tp_model_utils.py Signed-off-by: Wang, Yi A <yi.a.wang@intel.com> --------- Signed-off-by: Wang, Yi A <yi.a.wang@intel.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
Author
Parents
Loading