commit ff8864299a7896a9a41a34335f1ac8794a0d11cd Author: ModelHub XC Date: Fri May 22 03:24:12 2026 +0800 初始化项目,由ModelHub XC社区提供模型 Model: ibm-research/PowerMoE-3b Source: Original Platform diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..7636a52 --- /dev/null +++ b/.gitattributes @@ -0,0 +1,52 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bin.* filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zstandard filter=lfs diff=lfs merge=lfs -text +*.tfevents* filter=lfs diff=lfs merge=lfs -text +*.db* filter=lfs diff=lfs merge=lfs -text +*.ark* filter=lfs diff=lfs merge=lfs -text +**/*ckpt*data* filter=lfs diff=lfs merge=lfs -text +**/*ckpt*.meta filter=lfs diff=lfs merge=lfs -text +**/*ckpt*.index filter=lfs diff=lfs merge=lfs -text + +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.gguf* filter=lfs diff=lfs merge=lfs -text +*.ggml filter=lfs diff=lfs merge=lfs -text +*.llamafile* filter=lfs diff=lfs merge=lfs -text +*.pt2 filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text + +tokenizer.json filter=lfs diff=lfs merge=lfs -text +model-00003-of-00003.safetensors filter=lfs diff=lfs merge=lfs -text +model-00001-of-00003.safetensors filter=lfs diff=lfs merge=lfs -text +model-00002-of-00003.safetensors filter=lfs diff=lfs merge=lfs -text \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..96f2d75 --- /dev/null +++ b/README.md @@ -0,0 +1,154 @@ +--- +pipeline_tag: text-generation +inference: false +license: apache-2.0 +library_name: transformers +model-index: +- name: ibm/PowerMoE-3b + results: + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: ARC + metrics: + - name: accuracy-norm + type: accuracy-norm + value: 58.1 + verified: false + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: BoolQ + metrics: + - name: accuracy + type: accuracy + value: 65.0 + verified: false + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: Hellaswag + metrics: + - name: accuracy-norm + type: accuracy-norm + value: 71.5 + verified: false + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: OpenBookQA + metrics: + - name: accuracy-norm + type: accuracy-norm + value: 41.0 + verified: false + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: PIQA + metrics: + - name: accuracy-norm + type: accuracy-norm + value: 79.1 + verified: false + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: Winogrande + metrics: + - name: accuracy-norm + type: accuracy-norm + value: 65.0 + verified: false + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: MMLU (5 shot) + metrics: + - name: accuracy + type: accuracy + value: 42.8 + verified: false + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: GSM8k (5 shot) + metrics: + - name: accuracy + type: accuracy + value: 25.9 + verified: false + - task: + type: text-generation + dataset: + type: lm-eval-harness + name: math (4 shot) + metrics: + - name: accuracy + type: accuracy + value: 14.8 + verified: false + - task: + type: text-generation + dataset: + type: bigcode-eval + name: humaneval + metrics: + - name: pass@1 + type: pass@1 + value: 20.1 + verified: false + - task: + type: text-generation + dataset: + type: bigcode-eval + name: MBPP + metrics: + - name: pass@1 + type: pass@1 + value: 32.4 + verified: false +--- + +## Model Summary +PowerMoE-3B is a 3B sparse Mixture-of-Experts (sMoE) language model trained with the Power learning rate scheduler. It sparsely activates 800M parameters for each token. It is trained on a mix of open-source and proprietary datasets. PowerMoE-3B has shown promising results compared to other dense models with 2x activate parameters across various benchmarks, including natural language multi-choices, code generation, and math reasoning. +Paper: https://arxiv.org/abs/2408.13359 + +## Usage +Note: Requires installing HF transformers from source. + +### Generation +This is a simple example of how to use **PowerMoE-3b** model. + +```python +import torch +from transformers import AutoModelForCausalLM, AutoTokenizer +device = "cuda" # or "cpu" +model_path = "ibm/PowerMoE-3b" +tokenizer = AutoTokenizer.from_pretrained(model_path) +# drop device_map if running on CPU +model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device) +model.eval() +# change input text as desired +prompt = "Write a code to find the maximum value in a list of numbers." +# tokenize the text +input_tokens = tokenizer(prompt, return_tensors="pt") +# transfer tokenized inputs to the device +for i in input_tokens: + input_tokens[i] = input_tokens[i].to(device) +# generate output tokens +output = model.generate(**input_tokens, max_new_tokens=100) +# decode output tokens into text +output = tokenizer.batch_decode(output) +# loop over the batch to print, in this example the batch size is 1 +for i in output: + print(i) +``` \ No newline at end of file diff --git a/config.json b/config.json new file mode 100644 index 0000000..16e897e --- /dev/null +++ b/config.json @@ -0,0 +1,36 @@ +{ + "activation_function": "silu", + "architectures": [ + "GraniteMoeForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "attention_multiplier": 0.015625, + "bos_token_id": 0, + "embedding_multiplier": 12.0, + "eos_token_id": 0, + "hidden_size": 1536, + "initializer_range": 0.02, + "intermediate_size": 512, + "logits_scaling": 6.0, + "max_position_embeddings": 4096, + "model_type": "granitemoe", + "num_attention_heads": 24, + "num_experts_per_tok": 8, + "num_hidden_layers": 32, + "num_key_value_heads": 8, + "num_local_experts": 40, + "output_router_logits": false, + "pad_token_id": 0, + "residual_multiplier": 0.22, + "rms_norm_eps": 1e-06, + "rope_scaling": null, + "rope_theta": 10000, + "router_aux_loss_coef": 0.001, + "router_jitter_noise": 0.0, + "tie_word_embeddings": true, + "torch_dtype": "float32", + "transformers_version": "4.44.0.dev0", + "use_cache": true, + "vocab_size": 49152 +} diff --git a/configuration.json b/configuration.json new file mode 100644 index 0000000..bbeeda1 --- /dev/null +++ b/configuration.json @@ -0,0 +1 @@ +{"framework": "pytorch", "task": "text-generation", "allow_remote": true} \ No newline at end of file diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..51ffea4 --- /dev/null +++ b/generation_config.json @@ -0,0 +1,7 @@ +{ + "_from_model_config": true, + "bos_token_id": 0, + "eos_token_id": 0, + "pad_token_id": 0, + "transformers_version": "4.44.0.dev0" +} diff --git a/model-00001-of-00003.safetensors b/model-00001-of-00003.safetensors new file mode 100644 index 0000000..8ed50cd --- /dev/null +++ b/model-00001-of-00003.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:410fe4e5985463497cfab62f2a7e1d1399ca33ea6030966187f106d76b494b5c +size 4759191760 diff --git a/model-00002-of-00003.safetensors b/model-00002-of-00003.safetensors new file mode 100644 index 0000000..b7685a0 --- /dev/null +++ b/model-00002-of-00003.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2c318c4e0cb92c1385968a9c03e7a7c1ba4b536d47dab7ab37e2aaf50f221009 +size 4834947880 diff --git a/model-00003-of-00003.safetensors b/model-00003-of-00003.safetensors new file mode 100644 index 0000000..b225f73 --- /dev/null +++ b/model-00003-of-00003.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:948cbf3057a50c450d6322bcbaf1e9adb8cdd22915010aee14a58ebd5fcbdf3a +size 3903040856 diff --git a/model.safetensors.index.json b/model.safetensors.index.json new file mode 100644 index 0000000..c3fbf7f --- /dev/null +++ b/model.safetensors.index.json @@ -0,0 +1,298 @@ +{ + "metadata": { + "total_size": 13497145344 + }, + "weight_map": { + "lm_head.weight": "model-00003-of-00003.safetensors", + "model.embed_tokens.weight": "model-00001-of-00003.safetensors", + "model.layers.0.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.0.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.0.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.1.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.1.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.1.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.10.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.10.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.10.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.11.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.11.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.11.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.11.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.12.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.12.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.12.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.12.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.13.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.13.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.13.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.13.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.14.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.14.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.14.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.14.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.15.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.15.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.15.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.15.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.16.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.16.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.16.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.16.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.17.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.17.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.17.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.17.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.18.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.18.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.18.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.18.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.19.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.19.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.19.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.19.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.2.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.2.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.2.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.20.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.20.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.20.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.21.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.21.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.21.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.22.block_sparse_moe.input_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.22.block_sparse_moe.output_linear.weight": "model-00002-of-00003.safetensors", + "model.layers.22.block_sparse_moe.router.layer.weight": "model-00002-of-00003.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.23.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.23.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.23.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.23.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", + "model.layers.24.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.24.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.24.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.24.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.25.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.25.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.25.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.25.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.26.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.26.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.26.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.26.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.27.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.27.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.27.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.27.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.28.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.28.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.28.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.28.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.28.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.28.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.28.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.28.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.28.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.29.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.29.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.29.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.29.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.29.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.29.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.29.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.29.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.29.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.3.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.3.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.3.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.30.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.30.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.30.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.30.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.30.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.30.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.30.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.30.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.30.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.31.block_sparse_moe.input_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.31.block_sparse_moe.output_linear.weight": "model-00003-of-00003.safetensors", + "model.layers.31.block_sparse_moe.router.layer.weight": "model-00003-of-00003.safetensors", + "model.layers.31.input_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.31.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", + "model.layers.31.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.31.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.31.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.31.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", + "model.layers.4.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.4.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.4.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.5.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.5.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.5.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.6.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.6.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.6.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.7.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.7.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.7.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.8.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.8.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.8.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.9.block_sparse_moe.input_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.9.block_sparse_moe.output_linear.weight": "model-00001-of-00003.safetensors", + "model.layers.9.block_sparse_moe.router.layer.weight": "model-00001-of-00003.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", + "model.norm.weight": "model-00003-of-00003.safetensors" + } +} diff --git a/special_tokens_map.json b/special_tokens_map.json new file mode 100644 index 0000000..87a2385 --- /dev/null +++ b/special_tokens_map.json @@ -0,0 +1,45 @@ +{ + "additional_special_tokens": [ + "<|endoftext|>", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "" + ], + "bos_token": { + "content": "<|endoftext|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|endoftext|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": "<|endoftext|>", + "unk_token": { + "content": "<|endoftext|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/tokenizer.json b/tokenizer.json new file mode 100644 index 0000000..21b17fe --- /dev/null +++ b/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d3b4df07a0ce3940b15c77e2ea17ab5627ca0c4bf982d2eb37966aea0a81c918 +size 2057451 diff --git a/tokenizer_config.json b/tokenizer_config.json new file mode 100644 index 0000000..e72c686 --- /dev/null +++ b/tokenizer_config.json @@ -0,0 +1,187 @@ +{ + "add_prefix_space": false, + "added_tokens_decoder": { + "0": { + "content": "<|endoftext|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "1": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "2": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "3": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "4": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "5": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "6": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "7": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "8": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "9": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "10": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "11": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "12": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "13": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "14": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "15": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "16": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "17": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "18": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + } + }, + "additional_special_tokens": [ + "<|endoftext|>", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "", + "" + ], + "bos_token": "<|endoftext|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|endoftext|>", + "model_max_length": 9223372036854775807, + "pad_token": "<|endoftext|>", + "padding_side": "left", + "tokenizer_class": "GPT2Tokenizer", + "unk_token": "<|endoftext|>", + "vocab_size": 49152 +}