github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-24 10:41:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'coati/support-pp' of github.com:TongLi3701/ColossalAI into coati/support-pp

Browse Source
This commit is contained in:
Tong Li 2024-08-12 11:28:16 +00:00
commit 56fd2dc5d2
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