From 75756dff9699cfe5da3b44379519e5674706b23d Mon Sep 17 00:00:00 2001 From: Amy <1008611@qq.com> Date: Mon, 22 Sep 2025 10:27:31 +0800 Subject: [PATCH] =?UTF-8?q?=E5=86=85=E5=AE=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 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" + } +}