DeepSpeed

5e09acea - Pass base_dir to model files can be loaded for auto-tp/meta-tensor.