apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-01 23:02:36 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Code Simplification] fix init_distributed_environment() (#2982)

Browse Source
This commit is contained in:
lizexu123
2025-07-24 11:43:28 +08:00
committed by GitHub
parent bfeb664ab8
commit 832d25334a
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/model_executor/layers/linear.py
View File

@@ -500,7 +500,7 @@ class QKVParallelLinear(ColumnParallelLinear):
k_bias = get_tensor(state_dict.pop(k_bias_key))
v_bias = get_tensor(state_dict.pop(v_bias_key))
qkv_bias = paddle.concat([q_bias, k_bias, v_bias], axis=-1)
self.bias.set_value(qkv_bias)
self.bias.set_value(qkv_bias)
class RowParallelLinear(LinearBase):