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

Signed-off-by: wchen61 <wchen61@foxmail.com>
This commit is contained in:
wchen61 2024-11-15 16:52:20 +08:00 committed by GitHub
parent 02dbf30e9a
commit 3d158cdc8d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -250,6 +250,9 @@ class FalconDecoderLayer(nn.Module):
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