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

remove unused parameter

Browse Source
This commit is contained in:
Tong Li 2025-06-19 07:14:16 +00:00
parent 8880b83791
commit c7d3d0dc8f
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/shardformer/modeling/qwen2.py
View File

@ -824,7 +824,6 @@ def get_lm_forward_with_dist_cross_entropy(shard_config: ShardConfig):
output_attentions=output_attentions,
output_hidden_states=output_hidden_states,
return_dict=return_dict,
force_sp_output_gather=False,
)
hidden_states = outputs[0]