[Bugfix]fix and needs_scalar_to_array logic check (#6238)

Co-authored-by: Robert Shaw <114415538+robertgshaw2-neuralmagic@users.noreply.github.com>
This commit is contained in:
Baoyuan Qi
2024-07-10 07:43:24 +08:00
committed by GitHub
parent 673dd4cae9
commit d3a245138a

View File

@@ -387,7 +387,7 @@ class MergedColumnParallelLinear(ColumnParallelLinear):
if loaded_shard_id is None:
# Loaded weight is already fused on disk (qkv/mlp).
if output_dim is None:
if needs_scalar_to_array is not None:
if needs_scalar_to_array:
param_data, loaded_weight = adjust_scalar_to_fused_array(
param_data, loaded_weight, 0)
@@ -549,7 +549,7 @@ class QKVParallelLinear(ColumnParallelLinear):
if loaded_shard_id is None:
# Loaded weight is already fused on disk (qkv/mlp).
if output_dim is None:
if needs_scalar_to_array is not None:
if needs_scalar_to_array:
param_data, loaded_weight = adjust_scalar_to_fused_array(
param_data, loaded_weight, 0)