初始化项目,由ModelHub XC社区提供模型

Model: Shanghai_AI_Laboratory/internlm2_5-7b-chat-4bit
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-04-20 11:01:05 +08:00
commit 6f636fca63
15 changed files with 2635 additions and 0 deletions

38
.gitattributes vendored Normal file
View File

@@ -0,0 +1,38 @@
*.7z filter=lfs diff=lfs merge=lfs -text
*.arrow filter=lfs diff=lfs merge=lfs -text
*.bin filter=lfs diff=lfs merge=lfs -text
*.bz2 filter=lfs diff=lfs merge=lfs -text
*.ckpt 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
*.mlmodel filter=lfs diff=lfs merge=lfs -text
*.model filter=lfs diff=lfs merge=lfs -text
*.msgpack filter=lfs diff=lfs merge=lfs -text
*.npy filter=lfs diff=lfs merge=lfs -text
*.npz 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
*.pickle filter=lfs diff=lfs merge=lfs -text
*.pkl 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
*.safetensors filter=lfs diff=lfs merge=lfs -text
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
*.tar.* 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
*.wasm filter=lfs diff=lfs merge=lfs -text
*.xz filter=lfs diff=lfs merge=lfs -text
*.zip filter=lfs diff=lfs merge=lfs -text
*.zst filter=lfs diff=lfs merge=lfs -text
*tfevents* filter=lfs diff=lfs merge=lfs -text
pytorch_model-00001-of-00003.bin filter=lfs diff=lfs merge=lfs -text
pytorch_model-00002-of-00003.bin filter=lfs diff=lfs merge=lfs -text
pytorch_model-00003-of-00003.bin filter=lfs diff=lfs merge=lfs -text

68
README.md Normal file
View File

@@ -0,0 +1,68 @@
---
license: apache-2.0
pipeline_tag: text-generation
---
<div align="center">
<img src="https://raw.githubusercontent.com/InternLM/lmdeploy/0be9e7ab6fe9a066cfb0a09d0e0c8d2e28435e58/resources/lmdeploy-logo.svg" width="450"/>
</div>
# INT4 Weight-only Quantization and Deployment (W4A16)
LMDeploy adopts [AWQ](https://arxiv.org/abs/2306.00978) algorithm for 4bit weight-only quantization. By developed the high-performance cuda kernel, the 4bit quantized model inference achieves up to 2.4x faster than FP16.
LMDeploy supports the following NVIDIA GPU for W4A16 inference:
- Turing(sm75): 20 series, T4
- Ampere(sm80,sm86): 30 series, A10, A16, A30, A100
- Ada Lovelace(sm90): 40 series
Before proceeding with the quantization and inference, please ensure that lmdeploy is installed.
```shell
pip install lmdeploy[all]
```
This article comprises the following sections:
<!-- toc -->
- [Inference](#inference)
- [Evaluation](#evaluation)
- [Service](#service)
<!-- tocstop -->
## Inference
Trying the following codes, you can perform the batched offline inference with the quantized model:
```python
from lmdeploy import pipeline, TurbomindEngineConfig
engine_config = TurbomindEngineConfig(model_format='awq')
pipe = pipeline("internlm/internlm2_5-7b-chat-4bit", backend_config=engine_config)
response = pipe(["Hi, pls intro yourself", "Shanghai is"])
print(response)
```
For more information about the pipeline parameters, please refer to [here](https://github.com/InternLM/lmdeploy/blob/main/docs/en/inference/pipeline.md).
## Evaluation
Please overview [this guide](https://opencompass.readthedocs.io/en/latest/advanced_guides/evaluation_turbomind.html) about model evaluation with LMDeploy.
## Service
LMDeploy's `api_server` enables models to be easily packed into services with a single command. The provided RESTful APIs are compatible with OpenAI's interfaces. Below are an example of service startup:
```shell
lmdeploy serve api_server internlm/internlm2_5-7b-chat-4bit --backend turbomind --model-format awq
```
The default port of `api_server` is `23333`. After the server is launched, you can communicate with server on terminal through `api_client`:
```shell
lmdeploy serve api_client http://0.0.0.0:23333
```
You can overview and try out `api_server` APIs online by swagger UI at `http://0.0.0.0:23333`, or you can also read the API specification from [here](https://github.com/InternLM/lmdeploy/blob/main/docs/en/serving/restful_api.md).

41
config.json Normal file
View File

@@ -0,0 +1,41 @@
{
"_name_or_path": "/nvme/shared_data/InternLM/internlm2_5-7b-chat",
"architectures": [
"InternLM2ForCausalLM"
],
"attn_implementation": "eager",
"auto_map": {
"AutoConfig": "configuration_internlm2.InternLM2Config",
"AutoModel": "modeling_internlm2.InternLM2ForCausalLM",
"AutoModelForCausalLM": "modeling_internlm2.InternLM2ForCausalLM"
},
"bias": false,
"bos_token_id": 1,
"eos_token_id": 2,
"fp16": true,
"hidden_act": "silu",
"hidden_size": 4096,
"initializer_range": 0.02,
"intermediate_size": 14336,
"max_position_embeddings": 32768,
"model_type": "internlm2",
"num_attention_heads": 32,
"num_hidden_layers": 32,
"num_key_value_heads": 8,
"pad_token_id": 2,
"quantization_config": {
"bits": 4,
"group_size": 128,
"quant_method": "awq",
"version": "gemm",
"zero_point": true
},
"rms_norm_eps": 1e-05,
"rope_scaling": null,
"rope_theta": 1000000,
"tie_word_embeddings": false,
"torch_dtype": "float16",
"transformers_version": "4.40.1",
"use_cache": false,
"vocab_size": 92544
}

1
configuration.json Normal file
View File

@@ -0,0 +1 @@
{"framework": "pytorch", "task": "text-generation", "allow_remote": true}

151
configuration_internlm2.py Normal file
View File

@@ -0,0 +1,151 @@
# coding=utf-8
# Copyright (c) The InternLM team and The HuggingFace Inc. team. All rights reserved.
#
# This code is based on transformers/src/transformers/models/llama/configuration_llama.py
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
""" InternLM2 model configuration"""
from transformers.configuration_utils import PretrainedConfig
from transformers.utils import logging
logger = logging.get_logger(__name__)
INTERNLM2_PRETRAINED_CONFIG_ARCHIVE_MAP = {}
# Modified from transformers.model.llama.configuration_llama.LlamaConfig
class InternLM2Config(PretrainedConfig):
r"""
This is the configuration class to store the configuration of a [`InternLM2Model`]. It is used to instantiate
an InternLM2 model according to the specified arguments, defining the model architecture. Instantiating a
configuration with the defaults will yield a similar configuration to that of the InternLM2-7B.
Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
documentation from [`PretrainedConfig`] for more information.
Args:
vocab_size (`int`, *optional*, defaults to 32000):
Vocabulary size of the InternLM2 model. Defines the number of different tokens that can be represented by the
`inputs_ids` passed when calling [`InternLM2Model`]
hidden_size (`int`, *optional*, defaults to 4096):
Dimension of the hidden representations.
intermediate_size (`int`, *optional*, defaults to 11008):
Dimension of the MLP representations.
num_hidden_layers (`int`, *optional*, defaults to 32):
Number of hidden layers in the Transformer encoder.
num_attention_heads (`int`, *optional*, defaults to 32):
Number of attention heads for each attention layer in the Transformer encoder.
num_key_value_heads (`int`, *optional*):
This is the number of key_value heads that should be used to implement Grouped Query Attention. If
`num_key_value_heads=num_attention_heads`, the model will use Multi Head Attention (MHA), if
`num_key_value_heads=1 the model will use Multi Query Attention (MQA) otherwise GQA is used. When
converting a multi-head checkpoint to a GQA checkpoint, each group key and value head should be constructed
by meanpooling all the original heads within that group. For more details checkout [this
paper](https://arxiv.org/pdf/2305.13245.pdf). If it is not specified, will default to
`num_attention_heads`.
hidden_act (`str` or `function`, *optional*, defaults to `"silu"`):
The non-linear activation function (function or string) in the decoder.
max_position_embeddings (`int`, *optional*, defaults to 2048):
The maximum sequence length that this model might ever be used with. Typically set this to something large
just in case (e.g., 512 or 1024 or 2048).
initializer_range (`float`, *optional*, defaults to 0.02):
The standard deviation of the truncated_normal_initializer for initializing all weight matrices.
rms_norm_eps (`float`, *optional*, defaults to 1e-12):
The epsilon used by the rms normalization layers.
use_cache (`bool`, *optional*, defaults to `True`):
Whether or not the model should return the last key/values attentions (not used by all models). Only
relevant if `config.is_decoder=True`.
tie_word_embeddings(`bool`, *optional*, defaults to `False`):
Whether to tie weight embeddings
Example:
"""
model_type = "internlm2"
_auto_class = "AutoConfig"
def __init__( # pylint: disable=W0102
self,
vocab_size=103168,
hidden_size=4096,
intermediate_size=11008,
num_hidden_layers=32,
num_attention_heads=32,
num_key_value_heads=None,
hidden_act="silu",
max_position_embeddings=2048,
initializer_range=0.02,
rms_norm_eps=1e-6,
use_cache=True,
pad_token_id=0,
bos_token_id=1,
eos_token_id=2,
tie_word_embeddings=False,
bias=True,
rope_theta=10000,
rope_scaling=None,
attn_implementation="eager",
**kwargs,
):
self.vocab_size = vocab_size
self.max_position_embeddings = max_position_embeddings
self.hidden_size = hidden_size
self.intermediate_size = intermediate_size
self.num_hidden_layers = num_hidden_layers
self.num_attention_heads = num_attention_heads
self.bias = bias
if num_key_value_heads is None:
num_key_value_heads = num_attention_heads
self.num_key_value_heads = num_key_value_heads
self.hidden_act = hidden_act
self.initializer_range = initializer_range
self.rms_norm_eps = rms_norm_eps
self.use_cache = use_cache
self.rope_theta = rope_theta
self.rope_scaling = rope_scaling
self._rope_scaling_validation()
self.attn_implementation = attn_implementation
if self.attn_implementation is None:
self.attn_implementation = "eager"
super().__init__(
pad_token_id=pad_token_id,
bos_token_id=bos_token_id,
eos_token_id=eos_token_id,
tie_word_embeddings=tie_word_embeddings,
**kwargs,
)
def _rope_scaling_validation(self):
"""
Validate the `rope_scaling` configuration.
"""
if self.rope_scaling is None:
return
if not isinstance(self.rope_scaling, dict) or len(self.rope_scaling) != 2:
raise ValueError(
"`rope_scaling` must be a dictionary with with two fields, `type` and `factor`, "
f"got {self.rope_scaling}"
)
rope_scaling_type = self.rope_scaling.get("type", None)
rope_scaling_factor = self.rope_scaling.get("factor", None)
if rope_scaling_type is None or rope_scaling_type not in ["linear", "dynamic"]:
raise ValueError(
f"`rope_scaling`'s type field must be one of ['linear', 'dynamic'], got {rope_scaling_type}"
)
if rope_scaling_factor is None or not isinstance(rope_scaling_factor, float) or rope_scaling_factor < 1.0:
raise ValueError(f"`rope_scaling`'s factor field must be a float >= 1, got {rope_scaling_factor}")

7
generation_config.json Normal file
View File

@@ -0,0 +1,7 @@
{
"_from_model_config": true,
"bos_token_id": 1,
"eos_token_id": 2,
"pad_token_id": 2,
"transformers_version": "4.40.1"
}

1391
modeling_internlm2.py Normal file

File diff suppressed because it is too large Load Diff

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:7cc43082ea3b712911c7da2aa8f0cbf9a0a7bcbbc81669b20048409f04a67223
size 1974268315

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f6c91a01b2112daa4039c2976a89a8cfe199f702a429cb616e24710a7da0127d
size 1978993866

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:2a06a844d1c4db5a9aaf225c985618c55b11dc7a82510e6b2095b67624c7e28e
size 1189672234

View File

@@ -0,0 +1,554 @@
{
"metadata": {
"total_size": 5142749184
},
"weight_map": {
"model.layers.0.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.0.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.0.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.0.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.0.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.0.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.0.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.0.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.0.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.1.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.1.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.1.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.1.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.1.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.1.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.1.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.1.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.1.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.10.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.10.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.10.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.10.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.10.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.10.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.10.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.10.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.10.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.10.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.10.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.10.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.10.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.10.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.10.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.10.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.10.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.11.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.11.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.11.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.11.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.11.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.11.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.11.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.11.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.11.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.12.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.12.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.12.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.12.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.12.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.12.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.12.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.12.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.12.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.13.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.13.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.13.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.13.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.13.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.13.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.13.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.13.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.13.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.14.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.14.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.14.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.14.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.14.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.14.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.14.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.14.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.14.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.15.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.15.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.15.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.15.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.15.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.15.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.15.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.15.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.15.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.16.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.16.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.16.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.16.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.16.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.16.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.16.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.16.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.16.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.17.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.17.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.17.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.17.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.17.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.17.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.17.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.17.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.17.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.18.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.18.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.18.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.18.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.18.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.18.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.18.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.18.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.18.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.19.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.19.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.19.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.19.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.19.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.19.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.19.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.19.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.19.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.2.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.2.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.2.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.2.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.2.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.2.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.2.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.2.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.2.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.20.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.20.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.20.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.20.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.20.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.20.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.20.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.20.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.20.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.21.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.21.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.21.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.21.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.21.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.21.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.21.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.21.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.21.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.22.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.22.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.22.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.22.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.22.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.22.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.22.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.22.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.22.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.23.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.23.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.23.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.23.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.23.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.23.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.23.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.23.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.23.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.24.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.24.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.24.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.24.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.24.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.24.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.24.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.24.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.24.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.25.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.25.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.25.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.25.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.25.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.25.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.25.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.25.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.25.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.26.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.26.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.26.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.26.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.26.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.26.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.26.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.26.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.26.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.27.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.27.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.27.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.27.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.27.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.27.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.27.attention_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w1.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w1.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w1.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w2.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w2.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w2.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w3.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w3.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.27.feed_forward.w3.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.27.ffn_norm.weight": "pytorch_model-00002-of-00003.bin",
"model.layers.28.attention.wo.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.28.attention.wo.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.28.attention.wo.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.28.attention.wqkv.qweight": "pytorch_model-00002-of-00003.bin",
"model.layers.28.attention.wqkv.qzeros": "pytorch_model-00002-of-00003.bin",
"model.layers.28.attention.wqkv.scales": "pytorch_model-00002-of-00003.bin",
"model.layers.28.attention_norm.weight": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w1.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w1.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w1.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w2.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w2.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w2.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w3.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w3.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.28.feed_forward.w3.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.28.ffn_norm.weight": "pytorch_model-00003-of-00003.bin",
"model.layers.29.attention.wo.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.29.attention.wo.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.29.attention.wo.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.29.attention.wqkv.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.29.attention.wqkv.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.29.attention.wqkv.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.29.attention_norm.weight": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w1.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w1.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w1.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w2.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w2.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w2.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w3.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w3.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.29.feed_forward.w3.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.29.ffn_norm.weight": "pytorch_model-00003-of-00003.bin",
"model.layers.3.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.3.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.3.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.3.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.3.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.3.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.3.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.3.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.3.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.30.attention.wo.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.30.attention.wo.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.30.attention.wo.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.30.attention.wqkv.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.30.attention.wqkv.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.30.attention.wqkv.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.30.attention_norm.weight": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w1.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w1.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w1.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w2.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w2.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w2.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w3.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w3.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.30.feed_forward.w3.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.30.ffn_norm.weight": "pytorch_model-00003-of-00003.bin",
"model.layers.31.attention.wo.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.31.attention.wo.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.31.attention.wo.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.31.attention.wqkv.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.31.attention.wqkv.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.31.attention.wqkv.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.31.attention_norm.weight": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w1.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w1.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w1.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w2.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w2.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w2.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w3.qweight": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w3.qzeros": "pytorch_model-00003-of-00003.bin",
"model.layers.31.feed_forward.w3.scales": "pytorch_model-00003-of-00003.bin",
"model.layers.31.ffn_norm.weight": "pytorch_model-00003-of-00003.bin",
"model.layers.4.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.4.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.4.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.4.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.4.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.4.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.4.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.4.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.4.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.5.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.5.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.5.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.5.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.5.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.5.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.5.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.5.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.5.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.6.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.6.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.6.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.6.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.6.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.6.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.6.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.6.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.6.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.7.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.7.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.7.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.7.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.7.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.7.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.7.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.7.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.7.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.8.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.8.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.8.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.8.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.8.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.8.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.8.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.8.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.8.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.9.attention.wo.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.9.attention.wo.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.9.attention.wo.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.9.attention.wqkv.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.9.attention.wqkv.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.9.attention.wqkv.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.9.attention_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w1.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w1.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w1.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w2.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w2.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w2.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w3.qweight": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w3.qzeros": "pytorch_model-00001-of-00003.bin",
"model.layers.9.feed_forward.w3.scales": "pytorch_model-00001-of-00003.bin",
"model.layers.9.ffn_norm.weight": "pytorch_model-00001-of-00003.bin",
"model.norm.weight": "pytorch_model-00003-of-00003.bin",
"model.tok_embeddings.weight": "pytorch_model-00001-of-00003.bin",
"output.weight": "pytorch_model-00003-of-00003.bin"
}
}

38
special_tokens_map.json Normal file
View File

@@ -0,0 +1,38 @@
{
"additional_special_tokens": [
"<|im_start|>",
"<|im_end|>",
"<|action_start|>",
"<|action_end|>",
"<|interpreter|>",
"<|plugin|>"
],
"bos_token": {
"content": "<s>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false
},
"eos_token": {
"content": "</s>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false
},
"pad_token": {
"content": "</s>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false
},
"unk_token": {
"content": "<unk>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false
}
}

236
tokenization_internlm2.py Normal file
View File

@@ -0,0 +1,236 @@
# coding=utf-8
# Copyright (c) The InternLM team and The HuggingFace Inc. team. All rights reserved.
#
# This code is based on transformers/src/transformers/models/llama/tokenization_llama.py
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Tokenization classes for InternLM."""
import os
from shutil import copyfile
from typing import Any, Dict, List, Optional, Tuple
import sentencepiece as spm
from transformers.tokenization_utils import PreTrainedTokenizer
from transformers.utils import logging
logger = logging.get_logger(__name__)
VOCAB_FILES_NAMES = {"vocab_file": "./tokenizer.model"}
PRETRAINED_VOCAB_FILES_MAP = {}
# Modified from transformers.model.llama.tokenization_llama.LlamaTokenizer
class InternLM2Tokenizer(PreTrainedTokenizer):
"""
Construct a InternLM2 tokenizer. Based on byte-level Byte-Pair-Encoding.
Args:
vocab_file (`str`):
Path to the vocabulary file.
"""
vocab_files_names = VOCAB_FILES_NAMES
pretrained_vocab_files_map = PRETRAINED_VOCAB_FILES_MAP
model_input_names = ["input_ids", "attention_mask"]
_auto_class = "AutoTokenizer"
def __init__(
self,
vocab_file,
unk_token="<unk>",
bos_token="<s>",
eos_token="</s>",
pad_token="</s>",
sp_model_kwargs: Optional[Dict[str, Any]] = None,
add_bos_token=True,
add_eos_token=False,
decode_with_prefix_space=False,
clean_up_tokenization_spaces=False,
**kwargs,
):
self.sp_model_kwargs = {} if sp_model_kwargs is None else sp_model_kwargs
self.vocab_file = vocab_file
self.add_bos_token = add_bos_token
self.add_eos_token = add_eos_token
self.decode_with_prefix_space = decode_with_prefix_space
self.sp_model = spm.SentencePieceProcessor(**self.sp_model_kwargs)
self.sp_model.Load(vocab_file)
self._no_prefix_space_tokens = None
super().__init__(
bos_token=bos_token,
eos_token=eos_token,
unk_token=unk_token,
pad_token=pad_token,
clean_up_tokenization_spaces=clean_up_tokenization_spaces,
**kwargs,
)
@property
def no_prefix_space_tokens(self):
if self._no_prefix_space_tokens is None:
vocab = self.convert_ids_to_tokens(list(range(self.vocab_size)))
self._no_prefix_space_tokens = {i for i, tok in enumerate(vocab) if not tok.startswith("")}
return self._no_prefix_space_tokens
@property
def vocab_size(self):
"""Returns vocab size"""
return self.sp_model.get_piece_size()
@property
def bos_token_id(self) -> Optional[int]:
return self.sp_model.bos_id()
@property
def eos_token_id(self) -> Optional[int]:
return self.sp_model.eos_id()
def get_vocab(self):
"""Returns vocab as a dict"""
vocab = {self.convert_ids_to_tokens(i): i for i in range(self.vocab_size)}
vocab.update(self.added_tokens_encoder)
return vocab
def _tokenize(self, text):
"""Returns a tokenized string."""
return self.sp_model.encode(text, out_type=str)
def _convert_token_to_id(self, token):
"""Converts a token (str) in an id using the vocab."""
return self.sp_model.piece_to_id(token)
def _convert_id_to_token(self, index):
"""Converts an index (integer) in a token (str) using the vocab."""
token = self.sp_model.IdToPiece(index)
return token
def _maybe_add_prefix_space(self, tokens, decoded):
if tokens and tokens[0] not in self.no_prefix_space_tokens:
return " " + decoded
else:
return decoded
def convert_tokens_to_string(self, tokens):
"""Converts a sequence of tokens (string) in a single string."""
current_sub_tokens = []
out_string = ""
prev_is_special = False
for token in tokens:
# make sure that special tokens are not decoded using sentencepiece model
if token in self.all_special_tokens:
if not prev_is_special:
out_string += " "
out_string += self.sp_model.decode(current_sub_tokens) + token
prev_is_special = True
current_sub_tokens = []
else:
current_sub_tokens.append(token)
prev_is_special = False
out_string += self.sp_model.decode(current_sub_tokens)
out_string = self.clean_up_tokenization(out_string)
out_string = self._maybe_add_prefix_space(tokens=tokens, decoded=out_string)
return out_string[1:]
def save_vocabulary(self, save_directory, filename_prefix: Optional[str] = None) -> Tuple[str]:
"""
Save the vocabulary and special tokens file to a directory.
Args:
save_directory (`str`):
The directory in which to save the vocabulary.
Returns:
`Tuple(str)`: Paths to the files saved.
"""
if not os.path.isdir(save_directory):
logger.error(f"Vocabulary path ({save_directory}) should be a directory")
return
out_vocab_file = os.path.join(
save_directory, (filename_prefix + "-" if filename_prefix else "") + VOCAB_FILES_NAMES["vocab_file"]
)
if os.path.abspath(self.vocab_file) != os.path.abspath(out_vocab_file) and os.path.isfile(self.vocab_file):
copyfile(self.vocab_file, out_vocab_file)
elif not os.path.isfile(self.vocab_file):
with open(out_vocab_file, "wb") as fi:
content_spiece_model = self.sp_model.serialized_model_proto()
fi.write(content_spiece_model)
return (out_vocab_file,)
def build_inputs_with_special_tokens(self, token_ids_0, token_ids_1=None):
if self.add_bos_token:
bos_token_ids = [self.bos_token_id]
else:
bos_token_ids = []
output = bos_token_ids + token_ids_0
if token_ids_1 is not None:
output = output + token_ids_1
if self.add_eos_token:
output = output + [self.eos_token_id]
return output
def get_special_tokens_mask(
self, token_ids_0: List[int], token_ids_1: Optional[List[int]] = None, already_has_special_tokens: bool = False
) -> List[int]:
"""
Retrieve sequence ids from a token list that has no special tokens added. This method is called when adding
special tokens using the tokenizer `prepare_for_model` method.
Args:
token_ids_0 (`List[int]`):
List of IDs.
token_ids_1 (`List[int]`, *optional*):
Optional second list of IDs for sequence pairs.
already_has_special_tokens (`bool`, *optional*, defaults to `False`):
Whether or not the token list is already formatted with special tokens for the model.
Returns:
`List[int]`: A list of integers in the range [0, 1]: 1 for a special token, 0 for a sequence token.
"""
if already_has_special_tokens:
return super().get_special_tokens_mask(
token_ids_0=token_ids_0, token_ids_1=token_ids_1, already_has_special_tokens=True
)
if token_ids_1 is None:
return [1] + ([0] * len(token_ids_0)) + [1]
return [1] + ([0] * len(token_ids_0)) + [1, 1] + ([0] * len(token_ids_1)) + [1]
def create_token_type_ids_from_sequences(
self, token_ids_0: List[int], token_ids_1: Optional[List[int]] = None
) -> List[int]:
"""
Create a mask from the two sequences passed to be used in a sequence-pair classification task. T5 does not make
use of token type ids, therefore a list of zeros is returned.
Args:
token_ids_0 (`List[int]`):
List of IDs.
token_ids_1 (`List[int]`, *optional*):
Optional second list of IDs for sequence pairs.
Returns:
`List[int]`: List of zeros.
"""
eos = [self.eos_token_id]
if token_ids_1 is None:
return len(token_ids_0 + eos) * [0]
return len(token_ids_0 + eos + token_ids_1 + eos) * [0]

3
tokenizer.model Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f868398fc4e05ee1e8aeba95ddf18ddcc45b8bce55d5093bead5bbf80429b48b
size 1477754

98
tokenizer_config.json Normal file
View File

@@ -0,0 +1,98 @@
{
"added_tokens_decoder": {
"0": {
"content": "<unk>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"1": {
"content": "<s>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"2": {
"content": "</s>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"92538": {
"content": "<|plugin|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"92539": {
"content": "<|interpreter|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"92540": {
"content": "<|action_end|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"92541": {
"content": "<|action_start|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"92542": {
"content": "<|im_end|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"92543": {
"content": "<|im_start|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
}
},
"additional_special_tokens": [
"<|im_start|>",
"<|im_end|>",
"<|action_start|>",
"<|action_end|>",
"<|interpreter|>",
"<|plugin|>"
],
"auto_map": {
"AutoTokenizer": [
"tokenization_internlm2.InternLM2Tokenizer",
null
]
},
"bos_token": "<s>",
"chat_template": "{{ bos_token }}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
"clean_up_tokenization_spaces": false,
"eos_token": "</s>",
"model_max_length": 1000000000000000019884624838656,
"pad_token": "</s>",
"tokenizer_class": "InternLM2Tokenizer",
"unk_token": "<unk>"
}