github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 20:40:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

[format] applied code formatting on changed files in pull request 5067 (#5072)

Browse Source 
Co-authored-by: github-actions <github-actions@github.com>
This commit is contained in:
github-actions[bot]
2023-11-20 19:46:43 +08:00
committed by GitHub
parent fb103cfd6e
commit 8921a73c90
6 changed files with 27 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/zero/gemini/gemini_ddp.py
View File

@@ -905,4 +905,4 @@ class GeminiDDP(ModelWrapper):
if block is not None:
yield block, block_size
yield sharder.current_block, sharder.current_block_size
yield sharder.current_block, sharder.current_block_size