diff --git a/optimum/neuron/distributed/utils.py b/optimum/neuron/distributed/utils.py index 54b8d0683..8d3664d2c 100644 --- a/optimum/neuron/distributed/utils.py +++ b/optimum/neuron/distributed/utils.py @@ -781,7 +781,6 @@ def maybe_load_linear_weight_to_gqa_qkv_column_parallel_linear( "A linear's layer WeightInformation or a linear layer to copy the weights from need to specified." ) - # proj_name = weight_name[-1] if layer.fuse_qkv: weight = getattr(layer, "weight_qkv") bias = getattr(layer, "bias_qkv")