Lev Kurilenko 75db3d7da7 Fix SD workflow to work with latest diffusers version (#4918) 9 月之前
..
autotuning ce535945e6 fix: change ==NONE to is (#3923) 1 年之前
checkpoint 8998707a2f Universal Checkpoint for Sequence Parallelism (#4752) 10 月之前
comm c37fe9cbfb Fix exception handling in get_all_ranks_from_group() function (#4862) 10 月之前
compression 389bf69319 fix: Remove duplicate word the (#4051) 1 年之前
elasticity 6b2365e4fa Re-enable elastic training for torch 2+ (#4010) 1 年之前
inference 834272531a Add support of Microsoft Phi-2 model to DeepSpeed-FastGen (#4812) 9 月之前
launcher 6d7b44a838 [NPU] load EXPORT_ENV based on different accelerators to support multi-node training on other devices (#4830) 10 月之前
model_implementations d5a7c1e0b4 Capture short kernel sequences to graph (#4318) 10 月之前
module_inject 75db3d7da7 Fix SD workflow to work with latest diffusers version (#4918) 9 月之前
moe 2afa1c7f2f Communication Optimization for Large-Scale Training (#4695) 11 月之前
monitor 604d701e35 Introduce pydantic_v1 compatibility module for pydantic>=2.0.0 support (#4407) 1 年之前
nebula cd4e473ee6 fix typo with deepspeed/ (#3547) 1 年之前
ops fd0a52c1ac use all_gather_into_tensor instead of all_gather (#4705) 10 月之前
pipe b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
profiling 61391229c9 Update flops profiler to recurse (#4374) 11 月之前
runtime d7b764e3d8 Unit tests for MiCS (#4792) 9 月之前
sequence 2afa1c7f2f Communication Optimization for Large-Scale Training (#4695) 11 月之前
utils 691458f8b6 zero_to_fp32.py: Handle a case where shape doesn't have numel attr (#4842) 9 月之前
__init__.py 57a27b0803 add type checker ignore to resolve that pylance can't resolved noqa annotation (#4102) 1 年之前
accelerator 9548d48f48 Abstract accelerator (step 2) (#2560) 1 年之前
constants.py 706a72562a Allow env var for timeout (#4405) 1 年之前
env_report.py c1ba6a104f [CANN] Support cpu offload optimizer for Ascend NPU (#4568) 11 月之前
git_version_info.py 57a27b0803 add type checker ignore to resolve that pylance can't resolved noqa annotation (#4102) 1 年之前
pydantic_v1.py 604d701e35 Introduce pydantic_v1 compatibility module for pydantic>=2.0.0 support (#4407) 1 年之前