Merge branch 'main' of github.com:duanjunwen/ColossalAI into hotfix/t…

…est_checkpoint_io
duanjunwen · Jun 13, 2024 · 0fa84a7 · 0fa84a7
2 parents c909c58 + b37ad1f
commit 0fa84a7
Show file tree

Hide file tree

Showing 30 changed files with 217 additions and 2,966 deletions.
diff --git a/applications/ColossalMoE/colossal_moe/__init__.py b/applications/ColossalMoE/colossal_moe/__init__.py
diff --git a/applications/ColossalMoE/colossal_moe/models/__init__.py b/applications/ColossalMoE/colossal_moe/models/__init__.py
diff --git a/applications/ColossalMoE/colossal_moe/models/mixtral_checkpoint.py b/applications/ColossalMoE/colossal_moe/models/mixtral_checkpoint.py
diff --git a/applications/ColossalMoE/colossal_moe/models/mixtral_layer.py b/applications/ColossalMoE/colossal_moe/models/mixtral_layer.py