mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-25 04:24:51 +00:00
remove unused parameter
This commit is contained in:
parent
8880b83791
commit
c7d3d0dc8f
@ -824,7 +824,6 @@ def get_lm_forward_with_dist_cross_entropy(shard_config: ShardConfig):
|
|||||||
output_attentions=output_attentions,
|
output_attentions=output_attentions,
|
||||||
output_hidden_states=output_hidden_states,
|
output_hidden_states=output_hidden_states,
|
||||||
return_dict=return_dict,
|
return_dict=return_dict,
|
||||||
force_sp_output_gather=False,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
hidden_states = outputs[0]
|
hidden_states = outputs[0]
|
||||||
|
Loading…
Reference in New Issue
Block a user