github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-26 15:32:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix save issue (#6279)

Browse Source 
Co-authored-by: Tong Li <tong.li35271158@gmail.com>
This commit is contained in:
Tong Li 2025-04-27 17:54:06 +08:00 committed by GitHub
parent 38008858e4
commit 28795f560c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
applications/ColossalChat/coati/distributed/grpo_consumer.py
View File

@ -62,6 +62,7 @@ class GRPOConsumer(BaseConsumer):
model_config, model_config,
plugin_config, plugin_config,
minibatch_size, minibatch_size,
save_interval=save_interval,
save_dir=save_dir, save_dir=save_dir,
) )
path = model_config.pop("path") path = model_config.pop("path")