Skip to content

Commit a1514fa

Browse files
committed
simplify class name
1 parent 3e8a401 commit a1514fa

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

fastdeploy/model_executor/layers/mtp_linear.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from .utils import get_tensor
2222

2323

24-
class ParallelEmbeddingHiddenStatesProjection(nn.Layer):
24+
class ParallelEHProjection(nn.Layer):
2525
"""
2626
"Parallelized Embedding Hidden States Projection.
2727
"""
@@ -45,7 +45,7 @@ def __init__(
4545
embedding_dim (int): size of hidden state.
4646
prefix (str): full name of the layer in the state dict
4747
"""
48-
super(ParallelEmbeddingHiddenStatesProjection, self).__init__()
48+
super(ParallelEHProjection, self).__init__()
4949
self.linear_weight_key = prefix + ".weight"
5050
if with_bias:
5151
self.linear_bias_key = prefix + ".bias"

fastdeploy/model_executor/models/ernie4_5_mtp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from paddleformers.utils.log import logger
2727

2828
from fastdeploy.config import FDConfig, ModelConfig
29-
from fastdeploy.model_executor.layers.mtp_linear import ParallelEmbeddingHiddenStatesProjection
29+
from fastdeploy.model_executor.layers.mtp_linear import ParallelEHProjection
3030
from fastdeploy.model_executor.layers.normalization import RMSNorm
3131
from fastdeploy.model_executor.models.ernie4_5_moe import Ernie4_5_DecoderLayer
3232
from fastdeploy.model_executor.models.model_base import ModelForCasualLM
@@ -286,7 +286,7 @@ def __init__(
286286
prefix="ernie.mtp_hidden_norm.0",
287287
)
288288

289-
self.eh_proj = ParallelEmbeddingHiddenStatesProjection(
289+
self.eh_proj = ParallelEHProjection(
290290
fd_config=fd_config,
291291
num_embeddings=fd_config.model_config.hidden_size,
292292
embedding_dim=fd_config.model_config.hidden_size * 2,

0 commit comments

Comments
 (0)