From b66a10fc37944c5fb98a0412054323038d6eb0cd Mon Sep 17 00:00:00 2001 From: Amy <1008611@qq.com> Date: Thu, 10 Jul 2025 10:35:13 +0800 Subject: [PATCH] neir --- config.json | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 config.json diff --git a/config.json b/config.json new file mode 100644 index 0000000..4acf191 --- /dev/null +++ b/config.json @@ -0,0 +1,31 @@ +{ + "architectures": [ + "BitNetForCausalLM" + ], + "auto_map": { + "AutoConfig": "configuration_bitnet.BitNetConfig", + "AutoModelForCausalLM": "modeling_bitnet.BitNetForCausalLM" + }, + "bos_token_id": 128000, + "eos_token_id": 128001, + "hidden_act": "relu2", + "hidden_size": 2560, + "initializer_range": 0.02, + "intermediate_size": 6912, + "max_position_embeddings": 4096, + "model_type": "bitnet", + "rms_norm_eps": 1e-05, + "num_attention_heads": 20, + "num_hidden_layers": 30, + "num_key_value_heads": 5, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "bfloat16", + "use_cache": true, + "vocab_size": 128256, + "quantization_config": { + "quant_method": "bitnet", + "linear_class": "autobitlinear", + "quantization_mode": "offline" + } +}