github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-23 18:39:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Gemini] make gemini usage simple (#1821)

Browse Source
This commit is contained in:
Jiarui Fang
2022-11-08 15:53:13 +08:00
committed by GitHub
parent 99870726b1
commit cd5a0d56fa
4 changed files with 49 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/nn/parallel/__init__.py
View File

@@ -1,3 +1,4 @@
from .data_parallel import ColoDDP, ZeroDDP
from .gemini_parallel import GeminiDDP
__all__ = ['ColoDDP', 'ZeroDDP']
__all__ = ['ColoDDP', 'ZeroDDP', 'GeminiDDP']