Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion paddleformers/transformers/ernie4_5/modeling.py
Original file line number Diff line number Diff line change
Expand Up @@ -984,7 +984,7 @@ def forward(
)

# if labels is None,means we need full output, instead of tensor_parallel_output
# tensor_parallel_output is togather with ParallelCrossEntropy
# tensor_parallel_output is together with ParallelCrossEntropy
logits = self.lm_head(hidden_states)

if return_dict: # aka Generate Decoding
Expand Down
4 changes: 2 additions & 2 deletions paddleformers/transformers/model_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2319,7 +2319,7 @@ def _fuse_or_split_keys(
base_model_prefix=prefix,
)
if post_quantize:
# Split -> quantize(Not support mdoel save)
# Split -> quantize(Not support model save)
state_dict = load_state_dict(
shard_file,
tp_actions if pre_tensor_parallel_split else None,
Expand All @@ -2334,7 +2334,7 @@ def _fuse_or_split_keys(
dtype,
)
else:
# quantize -> split(Support mdoel save)
# quantize -> split(Support model save)
state_dict = load_state_dict(
shard_file,
tp_actions if pre_tensor_parallel_split else None,
Expand Down
2 changes: 1 addition & 1 deletion tests/nn/test_linear.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ def _test_create_parallel_linear(config):

assert isinstance(linear, mpu.RowParallelLinear)
config.sequence_parallel = False
# Test createing squence row parallel linear
# Test creating sequence row parallel linear
linear = Linear.create(
in_features=config.in_features,
out_features=config.out_features,
Expand Down
2 changes: 1 addition & 1 deletion tests/transformers/test_configuration_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,7 +215,7 @@ class ConfigTestUtils:
def test_config_common_kwargs_is_complete(self):
base_config = PretrainedConfig()
missing_keys = [key for key in base_config.__dict__ if key not in config_common_kwargs]
# If this part of the test fails, you have arguments to addin config_common_kwargs above.
# If this part of the test fails, you have arguments to add in config_common_kwargs above.
self.assertListEqual(
missing_keys,
[
Expand Down