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

fix update

Browse Source
This commit is contained in:
Tong Li
2025-06-27 09:52:50 +08:00
parent 58cb4fb4f7
commit a0f8680e85
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
applications/ColossalChat/rl_example.py
View File

@@ -248,7 +248,7 @@ if __name__ == "__main__":
) )
if args.enable_profiling: if args.enable_profiling:
# If profiling is enabled, we force model to generate to max_new_tokens # If profiling is enabled, we force model to generate to max_new_tokens
inference_model_config.update( generate_config.update(
dict( dict(
max_tokens=args.max_new_tokens, # max new tokens max_tokens=args.max_new_tokens, # max new tokens
ignore_eos=True, ignore_eos=True,