github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-22 17:58:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-08-12 10:15:34 +00:00
parent 4a541aa27c
commit 515f8e4a43
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
applications/ColossalChat/coati/trainer/base.py
View File

@ -16,8 +16,7 @@ from coati.experience_buffer import NaiveExperienceBuffer
from coati.experience_maker import Experience from coati.experience_maker import Experience
from torch.optim import Optimizer from torch.optim import Optimizer
from colossalai.booster import Booster from colossalai.booster import Booster, Plugin
from colossalai.booster import Plugin
from .utils import is_rank_0 from .utils import is_rank_0