Add default value to avoid Falcon crash (#5363) (#10347)

Signed-off-by: wchen61 <[email protected]>
vllm-project · Nov 15, 2024 · 3d158cd · 3d158cd
1 parent 02dbf30
commit 3d158cd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/vllm/model_executor/models/falcon.py b/vllm/model_executor/models/falcon.py
@@ -250,6 +250,9 @@ def __init__(
         self.mlp = FalconMLP(config, quant_config)
         self.config = config
 
+        if (not hasattr(config, "num_ln_in_parallel_attn")):
+            config.num_ln_in_parallel_attn = None
+
         if (config.num_ln_in_parallel_attn is None
                 and config.new_decoder_architecture):
             config.num_ln_in_parallel_attn = 2