Skip to content

Commit 08a1244

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 32b2148 commit 08a1244

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

applications/ColossalChat/coati/distributed/grpo_consumer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -530,4 +530,4 @@ def state_dict(self):
530530
model = self.policy_model.unwrap()
531531
state_dict = model.state_dict()
532532
state_dict["consumer_global_step"] = torch.tensor([self.global_step], device=self.device)
533-
return state_dict
533+
return state_dict

colossalai/shardformer/modeling/qwen3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ def qwen3_for_causal_lm_forward(
273273
hidden_states: Optional[torch.FloatTensor] = None,
274274
stage_index: Optional[List[int]] = None,
275275
shard_config: ShardConfig = None,
276-
**kwargs
276+
**kwargs,
277277
):
278278
r"""
279279
Args:

0 commit comments

Comments
 (0)