diff --git a/vllm/model_executor/models/whisper.py b/vllm/model_executor/models/whisper.py index 0c777e4a5..7462d9f6e 100644 --- a/vllm/model_executor/models/whisper.py +++ b/vllm/model_executor/models/whisper.py @@ -958,8 +958,8 @@ def _create_fake_bias_for_k_proj( So that the bias for k_proj in qkv_proj can be initialized with zeros. """ for name, weight in weights: + yield name, weight if name.endswith(fake_bias_key_name): bias = torch.zeros(weight.size(0)) bias_name = name.replace("weight", "bias") - yield from [(name, weight), (bias_name, bias)] - yield name, weight + yield bias_name, bias