Skip to content

Commit aba16b6

Browse files
committed
[LLM] fix sharding v1 reshard
1 parent 4213c30 commit aba16b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlenlp/trainer/utils/reshard/sharding_v1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def shard(node_model_state, model, optimizer):
2525
unwrapped_optimizer = unwrap_optimizer(optimizer, DygraphShardingOptimizer)
2626
if unwrapped_optimizer is not None:
2727
optimizer = unwrapped_optimizer
28-
assert not is_sharding_opt(optimizer)
28+
assert is_sharding_opt(optimizer)
2929
param2rank = optimizer._param2rank
3030

3131
def filter_func(key):

0 commit comments

Comments
 (0)