File tree Expand file tree Collapse file tree 1 file changed +0
-8
lines changed Expand file tree Collapse file tree 1 file changed +0
-8
lines changed Original file line number Diff line number Diff line change @@ -6669,14 +6669,6 @@ def set_gguf_parameters(self):
66696669 ## Validation ##
66706670 assert self .hparams .get ("hidden_act" ) in [None , "silu" ], "Only SILU activation supported"
66716671 assert self .d_inner % d_head == 0 , f"SSM inner size { self .d_inner } not a multiple of head dim { d_head } "
6672-
6673-
6674- # Add Falcon Mamba2 specific configuration
6675- self .gguf_writer .add_ssm_head_dim (self .hparams ["mamba_d_head" ])
6676- self .gguf_writer .add_ssm_inner_size (self .hparams ["mamba_d_ssm" ])
6677- self .gguf_writer .add_head_count (self .find_hparam (["num_attention_heads" ]))
6678- self .gguf_writer .add_key_length (self .hparams ["head_dim" ])
6679- self .gguf_writer .add_value_length (self .hparams ["head_dim" ])
66806672 self .gguf_writer .add_head_count_kv (self .find_hparam (["num_key_value_heads" ], optional = True ) or
66816673 self .find_hparam (["num_attention_heads" ]))
66826674
You can’t perform that action at this time.
0 commit comments