初始化项目,由ModelHub XC社区提供模型
Model: ibm-granite/granite-4.0-tiny-base-preview Source: Original Platform
This commit is contained in:
47
.gitattributes
vendored
Normal file
47
.gitattributes
vendored
Normal file
@@ -0,0 +1,47 @@
|
||||
*.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
|
||||
*.safetensors 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
|
||||
312
README.md
Normal file
312
README.md
Normal file
@@ -0,0 +1,312 @@
|
||||
---
|
||||
license: apache-2.0
|
||||
library_name: transformers
|
||||
tags:
|
||||
- language
|
||||
- granite-4.0
|
||||
---
|
||||
|
||||
[](https://mot.isitopen.ai/model/1169)
|
||||
|
||||
# Granite-4.0-Tiny-Base-Preview
|
||||
|
||||
**Model Summary:**
|
||||
|
||||
Granite-4.0-Tiny-Base-Preview is a 7B-parameter hybrid mixture-of-experts (MoE) language model featuring a 128k token context window. The architecture leverages Mamba-2, superimposed with a softmax attention for enhanced expressiveness, with no positional encoding for better length generalization.
|
||||
|
||||
|
||||
|
||||
|
||||
- **Developers:** Granite Team, IBM
|
||||
- **Website**: [Granite Docs](https://www.ibm.com/granite/docs/)
|
||||
- **Release Date**: May 2nd, 2025
|
||||
- **License:** [Apache 2.0](https://www.apache.org/licenses/LICENSE-2.0)
|
||||
|
||||
**Supported Languages:**
|
||||
English, German, Spanish, French, Japanese, Portuguese, Arabic, Czech, Italian, Korean, Dutch, and Chinese. Users may finetune Granite 4.0 models for languages beyond these 12 languages.
|
||||
|
||||
**Intended Use:**
|
||||
Prominent use cases of LLMs in text-to-text generation include summarization, text classification, extraction, question-answering, and other long-context tasks. All Granite Base models are able to handle these tasks as they were trained on a large amount of data from various domains. Moreover, they can serve as baseline to create specialized models for specific application scenarios.
|
||||
|
||||
**Installation:**
|
||||
You need to install transformer from source to use this checkpoint.
|
||||
<!-- This is a simple example of how to use Granite-4.0-Tiny-Base-Preview model. -->
|
||||
|
||||
<!-- Usage: Install transformer from source or use transformer version v4.45 to use this checkpoint. -->
|
||||
<!-- We have a huggingface PR which is yet to be merged. -->
|
||||
HuggingFace PR: https://github.com/huggingface/transformers/pull/37658
|
||||
|
||||
Install transformer from source: https://huggingface.co/docs/transformers/en/installation#install-from-source
|
||||
<!-- While the native support of this model in Hugging Face Transformers is pending ([PR](https://github.com/huggingface/transformers/pull/37658)), you need to install transformers from the following source to use this model:
|
||||
```shell
|
||||
git clone https://github.com/Ssukriti/transformers.git
|
||||
cd transformers
|
||||
git checkout granitemoe_hybrid_external_cleanup
|
||||
pip install -e .
|
||||
``` -->
|
||||
<!-- Install the following libraries:
|
||||
|
||||
```shell
|
||||
pip install torch torchvision torchaudio
|
||||
pip install accelerate
|
||||
pip install transformers
|
||||
``` -->
|
||||
**Generation:**
|
||||
After installation, copy the code snippet below to run the example.
|
||||
|
||||
```python
|
||||
from transformers import AutoModelForCausalLM, AutoTokenizer
|
||||
device = "auto"
|
||||
model_path = "ibm-granite/granite-4.0-tiny-base-preview"
|
||||
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
|
||||
input_text = "Where is the Thomas J. Watson Research Center located?"
|
||||
# tokenize the text
|
||||
input_tokens = tokenizer(input_text, return_tensors="pt").to(device)
|
||||
# generate output tokens
|
||||
output = model.generate(**input_tokens,
|
||||
max_length=4000)
|
||||
# decode output tokens into text
|
||||
output = tokenizer.batch_decode(output)
|
||||
# print output
|
||||
print(output)
|
||||
```
|
||||
|
||||
**Evaluation Results:**
|
||||
|
||||
<table>
|
||||
<caption><b>Comparison with 3.3 Base models</b><sup id="fnref1"><a href="#fn1">1</a></caption>
|
||||
<thead>
|
||||
<tr>
|
||||
<th style="text-align:left; background-color: #001d6c; color: white;">Models</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">ARC-Challenge</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">Hellaswag</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">MMLU</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">TruthfulQA</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">Winogrande</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">GSM8K</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">DROP</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">NQ</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">AGIEval</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">TriviaQA</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">Avg</th>
|
||||
</tr></thead>
|
||||
<tbody>
|
||||
<!-- <tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: #2D2D2D;">Granite-3.1-2B-Base</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">46.83</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">74.9</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">54.87</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">38.93</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">71.8</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">53.0</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">30.08</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">24.46</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">38.24</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">63.18</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">49.63</td>
|
||||
</tr> -->
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: #2D2D2D;">Granite-3.3-2B-Base</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;"> 47.49 </td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;"> 73.2 </td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;"> 54.33 </td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;"> 40.83 </td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;"> 70.4 </td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;"> 50.0 </td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;"> 32.552 </td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">24.36</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">38.78</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">63.22</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">49.52</td>
|
||||
</tr>
|
||||
<!-- <tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: #2D2D2D;">Granite-3.1-8B-Base</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">53.51</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">81.4</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">64.28</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">51.27</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">76.2</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">70.5</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">45.87</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">35.97</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">48.99</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">78.33</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">60.63</td>
|
||||
</tr> -->
|
||||
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: #2D2D2D;">Granite-3.3-8B-Base</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">50.84</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">80.1</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">63.89</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">52.15</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">74.4</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">59.0</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">36.14</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">36.5</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">49.3</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">78.18</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: #2D2D2D;">58.05</td>
|
||||
</tr>
|
||||
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #DAE8FF; color: black;"><b>Granite-4.0-Tiny-Base-Preview</b></td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">54.52</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">75.80</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">57.86</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">44.57</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">71.1</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">49.0</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">41.74</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">28.48</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">42.61</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">67.85</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">53.35</td>
|
||||
</tr>
|
||||
</tbody></table>
|
||||
|
||||
|
||||
**Model Architecture:**
|
||||
Granite-4.0-Tiny-Base-Preview is based on a decoder-only dense transformer architecture. Core components of this architecture are: GQA and NoPE, MLP with SwiGLU, RMSNorm, and shared input/output embeddings.
|
||||
<table>
|
||||
<thead>
|
||||
<tr>
|
||||
<th style="text-align:left; background-color: #001d6c; color: white;">Model</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">2B Dense</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">8B Dense</th>
|
||||
<th style="text-align:center; background-color: #001d6c; color: white;">Granite-4.0-Tiny-Base-Preview</th>
|
||||
</tr></thead>
|
||||
<tbody>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">Embedding size</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">2048</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">4096</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;"> 1536 </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">Number of layers</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">40</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">40</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">40</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">Attention head size</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;;">64</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">128</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">128</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">Number of attention heads</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">32</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">32</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">12</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">Number of KV heads</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">8</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">8</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">4</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">MLP hidden size</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">8192</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">12800</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">512</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">MLP activation</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">SwiGLU</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">SwiGLU</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">SwiGLU</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">Initialization std</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">0.1</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">0.1</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;"> 0.1 </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">Sequence length</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">128K</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">128K</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">128K</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;">Position embedding</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">RoPE</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">RoPE</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">None</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;"># Parameters</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">2.5B</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">8.1B</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">6.7B</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;"># Active parameters</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">2.5B</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">8.1B</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;"> 1B </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:left; background-color: #FFFFFF; color: black;"># Training tokens</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">12T</td>
|
||||
<td style="text-align:center; background-color: #FFFFFF; color: black;">12T</td>
|
||||
<td style="text-align:center; background-color: #DAE8FF; color: black;">2.5T</td>
|
||||
</tr>
|
||||
</tbody></table>
|
||||
|
||||
**Training Data:**
|
||||
This model is trained on a mix of open source and proprietary data following a two-stage training strategy.
|
||||
* Stage 1 data: The data for stage 1 is sourced from diverse domains, such as: web, code, academic sources, books, and math data.
|
||||
* Stage 2 data: The data for stage 2 comprises a curated mix of high-quality data from the same domains, plus multilingual and instruction data. The goal of this second training phase is to enhance the model’s performance on specific tasks.
|
||||
contains a recitation of the related paragraph before the answer.
|
||||
|
||||
**Infrastructure:**
|
||||
We train Granite 4.0 Language Models using IBM's super computing cluster, Blue Vela, which is outfitted with NVIDIA H100 GPUs. This cluster provides a scalable and efficient infrastructure for training our models over thousands of GPUs.
|
||||
|
||||
**Ethical Considerations and Limitations:**
|
||||
The use of Large Language Models involves risks and ethical considerations people must be aware of, including but not limited to: bias and fairness, misinformation, and autonomous decision-making. Granite-4.0-Tiny-Base-Preview model is not the exception in this regard. Even though this model is suited for multiple generative AI tasks, it has not undergone any safety alignment, there it may produce problematic outputs. Additionally, it remains uncertain whether smaller models might exhibit increased susceptibility to hallucination in generation scenarios by copying text verbatim from the training dataset due to their reduced sizes and memorization capacities. This aspect is currently an active area of research, and we anticipate more rigorous exploration, comprehension, and mitigations in this domain. Regarding ethics, a latent risk associated with all Large Language Models is their malicious utilization. We urge the community to use Granite-4.0-Tiny-Base-Preview model with ethical intentions and in a responsible way.
|
||||
|
||||
**Signature verification:**
|
||||
Model signing is an experimental feature with ongoing development, which might include breaking changes. We are releasing these capabilities to improve the integrity of our models for our security-conscious users and to facilitate feedback from the community.
|
||||
|
||||
Before trying to verify the signature, ensure that the tensor files have been downloaded with git-lfs and that no files have been added, removed, or modified in your local git checkout:
|
||||
|
||||
```bash
|
||||
git lfs fetch --all
|
||||
git lfs pull
|
||||
git lfs checkout
|
||||
```
|
||||
|
||||
Install the model_signing (v1.0.1) library with the following command:
|
||||
|
||||
```bash
|
||||
pip install 'model-signing==v1.1.1'
|
||||
```
|
||||
|
||||
|
||||
|
||||
Then verify the signature with the following command ensuring that the IBM identity 'granite.preview@ibm.com' was used for signing this model:
|
||||
|
||||
```bash
|
||||
python -m model_signing verify sigstore \
|
||||
--signature model.sig \
|
||||
--ignore-paths .git \
|
||||
--ignore-paths .gitattributes \
|
||||
--identity Granite.Preview@ibm.com \
|
||||
--identity_provider https://sigstore.verify.ibm.com/oauth2 \
|
||||
.
|
||||
```
|
||||
|
||||
**Resources**
|
||||
- ⭐️ Learn about the latest updates with Granite: https://www.ibm.com/granite
|
||||
- 📄 Get started with tutorials, best practices, and prompt engineering advice: https://www.ibm.com/granite/docs/
|
||||
- 💡 Learn about the latest Granite learning resources: https://ibm.biz/granite-learning-resources
|
||||
|
||||
<p><a href="#fnref1" title="Jump back to reference">[1]</a> Evaluated using <a href="https://github.com/allenai/olmes">OLMES</a></p>
|
||||
89
config.json
Normal file
89
config.json
Normal file
@@ -0,0 +1,89 @@
|
||||
{
|
||||
"architectures": [
|
||||
"GraniteMoeHybridForCausalLM"
|
||||
],
|
||||
"attention_bias": false,
|
||||
"attention_dropout": 0.0,
|
||||
"attention_multiplier": 0.0078125,
|
||||
"bos_token_id": 0,
|
||||
"embedding_multiplier": 12,
|
||||
"eos_token_id": 0,
|
||||
"hidden_act": "silu",
|
||||
"hidden_size": 1536,
|
||||
"init_method": "mup",
|
||||
"initializer_range": 0.1,
|
||||
"intermediate_size": 512,
|
||||
"layer_types": [
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"attention",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"attention",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"attention",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"attention",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba",
|
||||
"mamba"
|
||||
],
|
||||
"logits_scaling": 6,
|
||||
"mamba_chunk_size": 256,
|
||||
"mamba_conv_bias": true,
|
||||
"mamba_d_conv": 4,
|
||||
"mamba_d_head": 64,
|
||||
"mamba_d_state": 128,
|
||||
"mamba_expand": 2,
|
||||
"mamba_n_groups": 1,
|
||||
"mamba_n_heads": 48,
|
||||
"mamba_proj_bias": false,
|
||||
"max_position_embeddings": 131072,
|
||||
"model_type": "granitemoehybrid",
|
||||
"normalization_function": "rmsnorm",
|
||||
"num_attention_heads": 12,
|
||||
"num_experts_per_tok": 6,
|
||||
"num_hidden_layers": 40,
|
||||
"num_key_value_heads": 4,
|
||||
"num_local_experts": 62,
|
||||
"output_router_logits": false,
|
||||
"pad_token_id": 0,
|
||||
"position_embedding_type": "nope",
|
||||
"residual_multiplier": 0.22,
|
||||
"rms_norm_eps": 1e-05,
|
||||
"rope_scaling": null,
|
||||
"rope_theta": 10000,
|
||||
"router_aux_loss_coef": 0.0,
|
||||
"shared_intermediate_size": 1024,
|
||||
"tie_word_embeddings": true,
|
||||
"transformers_version": "4.52.0.dev0",
|
||||
"use_cache": true,
|
||||
"vocab_size": 50304
|
||||
}
|
||||
1
configuration.json
Normal file
1
configuration.json
Normal file
@@ -0,0 +1 @@
|
||||
{"framework": "pytorch", "task": "text-generation", "allow_remote": true}
|
||||
7
generation_config.json
Normal file
7
generation_config.json
Normal file
@@ -0,0 +1,7 @@
|
||||
{
|
||||
"_from_model_config": true,
|
||||
"bos_token_id": 0,
|
||||
"eos_token_id": 0,
|
||||
"pad_token_id": 0,
|
||||
"transformers_version": "4.52.0.dev0"
|
||||
}
|
||||
3
model-00001-of-00006.safetensors
Normal file
3
model-00001-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:57be0ebde7435317126232ee9b64f3bd35fb5f511a143550430f91ae7a8e52f2
|
||||
size 4916677832
|
||||
3
model-00002-of-00006.safetensors
Normal file
3
model-00002-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:b525caa43b6fd9011de3200abae3d2444bdfab38cd30ee2a5b5dcf10069f7126
|
||||
size 4640615064
|
||||
3
model-00003-of-00006.safetensors
Normal file
3
model-00003-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:873bf478522543127f549372b6a7b2b58301c53e2d161dee09c3e16632c01403
|
||||
size 4997280664
|
||||
3
model-00004-of-00006.safetensors
Normal file
3
model-00004-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:d2dd493f8779da5d7cc515b5068a3b2e0f9d87753e14c9ab22898c64e94cc829
|
||||
size 4880084840
|
||||
3
model-00005-of-00006.safetensors
Normal file
3
model-00005-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:cc9bfa543d415b483d2161911a48ea5b0a4dd597a3c130e85c064a3e7b42f6b3
|
||||
size 4997280664
|
||||
3
model-00006-of-00006.safetensors
Normal file
3
model-00006-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:7236571cf1d508871815a560d01bbdd583822e1705e1139d6264fd0a0d57fdfa
|
||||
size 2261316064
|
||||
593
model.safetensors.index.json
Normal file
593
model.safetensors.index.json
Normal file
@@ -0,0 +1,593 @@
|
||||
{
|
||||
"metadata": {
|
||||
"total_size": 26693187840
|
||||
},
|
||||
"weight_map": {
|
||||
"model.embed_tokens.weight": "model-00001-of-00006.safetensors",
|
||||
"model.norm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.block_sparse_moe.router.layer.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.block_sparse_moe.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.block_sparse_moe.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.shared_mlp.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.shared_mlp.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mamba.conv1d.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mamba.conv1d.bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mamba.in_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mamba.dt_bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mamba.A_log": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mamba.D": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mamba.out_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mamba.norm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.block_sparse_moe.router.layer.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.block_sparse_moe.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.block_sparse_moe.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.shared_mlp.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.shared_mlp.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mamba.conv1d.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mamba.conv1d.bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mamba.in_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mamba.dt_bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mamba.A_log": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mamba.D": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mamba.out_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mamba.norm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.block_sparse_moe.router.layer.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.block_sparse_moe.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.block_sparse_moe.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.shared_mlp.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.shared_mlp.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mamba.conv1d.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mamba.conv1d.bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mamba.in_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mamba.dt_bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mamba.A_log": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mamba.D": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mamba.out_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mamba.norm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.block_sparse_moe.router.layer.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.block_sparse_moe.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.block_sparse_moe.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.shared_mlp.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.shared_mlp.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mamba.conv1d.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mamba.conv1d.bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mamba.in_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mamba.dt_bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mamba.A_log": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mamba.D": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mamba.out_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mamba.norm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.block_sparse_moe.router.layer.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.block_sparse_moe.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.block_sparse_moe.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.shared_mlp.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.shared_mlp.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mamba.conv1d.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mamba.conv1d.bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mamba.in_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mamba.dt_bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mamba.A_log": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mamba.D": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mamba.out_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mamba.norm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.block_sparse_moe.router.layer.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.block_sparse_moe.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.block_sparse_moe.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.shared_mlp.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.shared_mlp.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.self_attn.q_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.self_attn.k_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.self_attn.v_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.self_attn.o_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.block_sparse_moe.router.layer.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.block_sparse_moe.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.block_sparse_moe.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.shared_mlp.input_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.shared_mlp.output_linear.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.mamba.conv1d.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.mamba.conv1d.bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.mamba.in_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.mamba.dt_bias": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.mamba.A_log": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.mamba.D": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.mamba.out_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.mamba.norm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.7.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.7.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.7.block_sparse_moe.router.layer.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.7.block_sparse_moe.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.block_sparse_moe.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.shared_mlp.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.shared_mlp.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mamba.conv1d.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mamba.conv1d.bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mamba.in_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mamba.dt_bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mamba.A_log": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mamba.D": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mamba.out_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mamba.norm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.block_sparse_moe.router.layer.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.block_sparse_moe.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.block_sparse_moe.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.shared_mlp.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.shared_mlp.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mamba.conv1d.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mamba.conv1d.bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mamba.in_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mamba.dt_bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mamba.A_log": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mamba.D": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mamba.out_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mamba.norm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.block_sparse_moe.router.layer.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.block_sparse_moe.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.block_sparse_moe.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.shared_mlp.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.shared_mlp.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mamba.conv1d.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mamba.conv1d.bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mamba.in_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mamba.dt_bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mamba.A_log": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mamba.D": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mamba.out_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mamba.norm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.block_sparse_moe.router.layer.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.block_sparse_moe.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.block_sparse_moe.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.shared_mlp.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.shared_mlp.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mamba.conv1d.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mamba.conv1d.bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mamba.in_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mamba.dt_bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mamba.A_log": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mamba.D": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mamba.out_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mamba.norm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.block_sparse_moe.router.layer.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.block_sparse_moe.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.block_sparse_moe.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.shared_mlp.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.shared_mlp.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.mamba.conv1d.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.mamba.conv1d.bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.mamba.in_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.mamba.dt_bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.mamba.A_log": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.mamba.D": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.mamba.out_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.mamba.norm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.block_sparse_moe.router.layer.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.block_sparse_moe.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.block_sparse_moe.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.shared_mlp.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.shared_mlp.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.mamba.conv1d.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.mamba.conv1d.bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.mamba.in_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.mamba.dt_bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.mamba.A_log": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.mamba.D": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.mamba.out_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.mamba.norm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.block_sparse_moe.router.layer.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.block_sparse_moe.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.block_sparse_moe.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.shared_mlp.input_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.shared_mlp.output_linear.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.mamba.conv1d.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.mamba.conv1d.bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.mamba.in_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.mamba.dt_bias": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.mamba.A_log": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.mamba.D": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.mamba.out_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.13.mamba.norm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.14.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.14.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.14.block_sparse_moe.router.layer.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.14.block_sparse_moe.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.block_sparse_moe.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.shared_mlp.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.shared_mlp.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mamba.conv1d.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mamba.conv1d.bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mamba.in_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mamba.dt_bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mamba.A_log": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mamba.D": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mamba.out_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mamba.norm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.block_sparse_moe.router.layer.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.block_sparse_moe.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.block_sparse_moe.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.shared_mlp.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.shared_mlp.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.self_attn.q_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.self_attn.k_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.self_attn.v_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.self_attn.o_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.block_sparse_moe.router.layer.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.block_sparse_moe.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.block_sparse_moe.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.shared_mlp.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.shared_mlp.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mamba.conv1d.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mamba.conv1d.bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mamba.in_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mamba.dt_bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mamba.A_log": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mamba.D": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mamba.out_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mamba.norm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.block_sparse_moe.router.layer.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.block_sparse_moe.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.block_sparse_moe.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.shared_mlp.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.shared_mlp.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.mamba.conv1d.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.mamba.conv1d.bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.mamba.in_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.mamba.dt_bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.mamba.A_log": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.mamba.D": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.mamba.out_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.mamba.norm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.block_sparse_moe.router.layer.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.block_sparse_moe.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.block_sparse_moe.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.shared_mlp.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.shared_mlp.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.mamba.conv1d.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.mamba.conv1d.bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.mamba.in_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.mamba.dt_bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.mamba.A_log": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.mamba.D": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.mamba.out_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.mamba.norm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.block_sparse_moe.router.layer.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.block_sparse_moe.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.block_sparse_moe.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.shared_mlp.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.shared_mlp.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.mamba.conv1d.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.mamba.conv1d.bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.mamba.in_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.mamba.dt_bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.mamba.A_log": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.mamba.D": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.mamba.out_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.19.mamba.norm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.block_sparse_moe.router.layer.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.block_sparse_moe.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.block_sparse_moe.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.shared_mlp.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.shared_mlp.output_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.mamba.conv1d.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.mamba.conv1d.bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.mamba.in_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.mamba.dt_bias": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.mamba.A_log": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.mamba.D": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.mamba.out_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.20.mamba.norm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.21.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.21.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.21.block_sparse_moe.router.layer.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.21.block_sparse_moe.input_linear.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.21.block_sparse_moe.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.shared_mlp.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.shared_mlp.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mamba.conv1d.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mamba.conv1d.bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mamba.in_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mamba.dt_bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mamba.A_log": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mamba.D": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mamba.out_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mamba.norm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.block_sparse_moe.router.layer.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.block_sparse_moe.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.block_sparse_moe.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.shared_mlp.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.shared_mlp.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mamba.conv1d.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mamba.conv1d.bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mamba.in_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mamba.dt_bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mamba.A_log": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mamba.D": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mamba.out_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mamba.norm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.block_sparse_moe.router.layer.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.block_sparse_moe.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.block_sparse_moe.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.shared_mlp.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.shared_mlp.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.mamba.conv1d.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.mamba.conv1d.bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.mamba.in_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.mamba.dt_bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.mamba.A_log": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.mamba.D": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.mamba.out_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.mamba.norm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.block_sparse_moe.router.layer.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.block_sparse_moe.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.block_sparse_moe.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.shared_mlp.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.shared_mlp.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.mamba.conv1d.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.mamba.conv1d.bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.mamba.in_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.mamba.dt_bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.mamba.A_log": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.mamba.D": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.mamba.out_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.mamba.norm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.block_sparse_moe.router.layer.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.block_sparse_moe.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.block_sparse_moe.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.shared_mlp.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.shared_mlp.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.self_attn.q_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.self_attn.k_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.self_attn.v_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.25.self_attn.o_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.block_sparse_moe.router.layer.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.block_sparse_moe.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.block_sparse_moe.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.shared_mlp.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.shared_mlp.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.mamba.conv1d.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.mamba.conv1d.bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.mamba.in_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.mamba.dt_bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.mamba.A_log": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.mamba.D": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.mamba.out_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.26.mamba.norm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.block_sparse_moe.router.layer.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.block_sparse_moe.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.block_sparse_moe.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.shared_mlp.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.shared_mlp.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.mamba.conv1d.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.mamba.conv1d.bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.mamba.in_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.mamba.dt_bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.mamba.A_log": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.mamba.D": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.mamba.out_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.27.mamba.norm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.block_sparse_moe.router.layer.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.block_sparse_moe.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.block_sparse_moe.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.shared_mlp.input_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.shared_mlp.output_linear.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.mamba.conv1d.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.mamba.conv1d.bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.mamba.in_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.mamba.dt_bias": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.mamba.A_log": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.mamba.D": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.mamba.out_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.28.mamba.norm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.29.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.29.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.29.block_sparse_moe.router.layer.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.29.block_sparse_moe.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.block_sparse_moe.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.shared_mlp.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.shared_mlp.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.mamba.conv1d.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.mamba.conv1d.bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.mamba.in_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.mamba.dt_bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.mamba.A_log": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.mamba.D": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.mamba.out_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.mamba.norm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.block_sparse_moe.router.layer.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.block_sparse_moe.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.block_sparse_moe.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.shared_mlp.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.shared_mlp.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.mamba.conv1d.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.mamba.conv1d.bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.mamba.in_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.mamba.dt_bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.mamba.A_log": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.mamba.D": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.mamba.out_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.30.mamba.norm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.block_sparse_moe.router.layer.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.block_sparse_moe.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.block_sparse_moe.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.shared_mlp.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.shared_mlp.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.mamba.conv1d.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.mamba.conv1d.bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.mamba.in_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.mamba.dt_bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.mamba.A_log": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.mamba.D": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.mamba.out_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.31.mamba.norm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.block_sparse_moe.router.layer.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.block_sparse_moe.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.block_sparse_moe.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.shared_mlp.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.shared_mlp.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.mamba.conv1d.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.mamba.conv1d.bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.mamba.in_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.mamba.dt_bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.mamba.A_log": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.mamba.D": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.mamba.out_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.32.mamba.norm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.block_sparse_moe.router.layer.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.block_sparse_moe.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.block_sparse_moe.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.shared_mlp.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.shared_mlp.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.mamba.conv1d.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.mamba.conv1d.bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.mamba.in_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.mamba.dt_bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.mamba.A_log": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.mamba.D": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.mamba.out_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.33.mamba.norm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.block_sparse_moe.router.layer.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.block_sparse_moe.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.block_sparse_moe.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.shared_mlp.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.shared_mlp.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.mamba.conv1d.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.mamba.conv1d.bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.mamba.in_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.mamba.dt_bias": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.mamba.A_log": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.mamba.D": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.mamba.out_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.34.mamba.norm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.block_sparse_moe.router.layer.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.block_sparse_moe.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.block_sparse_moe.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.shared_mlp.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.shared_mlp.output_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.self_attn.q_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.self_attn.k_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.self_attn.v_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.35.self_attn.o_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.36.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.36.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.36.block_sparse_moe.router.layer.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.36.block_sparse_moe.input_linear.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.36.block_sparse_moe.output_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.shared_mlp.input_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.shared_mlp.output_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.mamba.conv1d.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.mamba.conv1d.bias": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.mamba.in_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.mamba.dt_bias": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.mamba.A_log": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.mamba.D": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.mamba.out_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.36.mamba.norm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.input_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.post_attention_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.block_sparse_moe.router.layer.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.block_sparse_moe.input_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.block_sparse_moe.output_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.shared_mlp.input_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.shared_mlp.output_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.mamba.conv1d.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.mamba.conv1d.bias": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.mamba.in_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.mamba.dt_bias": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.mamba.A_log": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.mamba.D": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.mamba.out_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.37.mamba.norm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.input_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.post_attention_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.block_sparse_moe.router.layer.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.block_sparse_moe.input_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.block_sparse_moe.output_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.shared_mlp.input_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.shared_mlp.output_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.mamba.conv1d.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.mamba.conv1d.bias": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.mamba.in_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.mamba.dt_bias": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.mamba.A_log": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.mamba.D": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.mamba.out_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.38.mamba.norm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.input_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.post_attention_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.block_sparse_moe.router.layer.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.block_sparse_moe.input_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.block_sparse_moe.output_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.shared_mlp.input_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.shared_mlp.output_linear.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.mamba.conv1d.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.mamba.conv1d.bias": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.mamba.in_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.mamba.dt_bias": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.mamba.A_log": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.mamba.D": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.mamba.out_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.39.mamba.norm.weight": "model-00006-of-00006.safetensors"
|
||||
}
|
||||
}
|
||||
44
special_tokens_map.json
Normal file
44
special_tokens_map.json
Normal file
@@ -0,0 +1,44 @@
|
||||
{
|
||||
"additional_special_tokens": [
|
||||
"<|endoftext|>",
|
||||
"<fim_prefix>",
|
||||
"<fim_middle>",
|
||||
"<fim_suffix>",
|
||||
"<fim_pad>",
|
||||
"<filename>",
|
||||
"<gh_stars>",
|
||||
"<issue_start>",
|
||||
"<issue_comment>",
|
||||
"<issue_closed>",
|
||||
"<jupyter_start>",
|
||||
"<jupyter_text>",
|
||||
"<jupyter_code>",
|
||||
"<jupyter_output>",
|
||||
"<empty_output>",
|
||||
"<commit_before>",
|
||||
"<commit_msg>",
|
||||
"<commit_after>",
|
||||
"<reponame>"
|
||||
],
|
||||
"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
|
||||
},
|
||||
"unk_token": {
|
||||
"content": "<|endoftext|>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
}
|
||||
}
|
||||
244931
tokenizer.json
Normal file
244931
tokenizer.json
Normal file
File diff suppressed because it is too large
Load Diff
186
tokenizer_config.json
Normal file
186
tokenizer_config.json
Normal file
@@ -0,0 +1,186 @@
|
||||
{
|
||||
"add_prefix_space": false,
|
||||
"added_tokens_decoder": {
|
||||
"0": {
|
||||
"content": "<|endoftext|>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"1": {
|
||||
"content": "<fim_prefix>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"2": {
|
||||
"content": "<fim_middle>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"3": {
|
||||
"content": "<fim_suffix>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"4": {
|
||||
"content": "<fim_pad>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"5": {
|
||||
"content": "<filename>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"6": {
|
||||
"content": "<gh_stars>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"7": {
|
||||
"content": "<issue_start>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"8": {
|
||||
"content": "<issue_comment>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"9": {
|
||||
"content": "<issue_closed>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"10": {
|
||||
"content": "<jupyter_start>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"11": {
|
||||
"content": "<jupyter_text>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"12": {
|
||||
"content": "<jupyter_code>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"13": {
|
||||
"content": "<jupyter_output>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"14": {
|
||||
"content": "<empty_output>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"15": {
|
||||
"content": "<commit_before>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"16": {
|
||||
"content": "<commit_msg>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"17": {
|
||||
"content": "<commit_after>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"18": {
|
||||
"content": "<reponame>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
}
|
||||
},
|
||||
"additional_special_tokens": [
|
||||
"<|endoftext|>",
|
||||
"<fim_prefix>",
|
||||
"<fim_middle>",
|
||||
"<fim_suffix>",
|
||||
"<fim_pad>",
|
||||
"<filename>",
|
||||
"<gh_stars>",
|
||||
"<issue_start>",
|
||||
"<issue_comment>",
|
||||
"<issue_closed>",
|
||||
"<jupyter_start>",
|
||||
"<jupyter_text>",
|
||||
"<jupyter_code>",
|
||||
"<jupyter_output>",
|
||||
"<empty_output>",
|
||||
"<commit_before>",
|
||||
"<commit_msg>",
|
||||
"<commit_after>",
|
||||
"<reponame>"
|
||||
],
|
||||
"bos_token": "<|endoftext|>",
|
||||
"clean_up_tokenization_spaces": false,
|
||||
"eos_token": "<|endoftext|>",
|
||||
"extra_special_tokens": {},
|
||||
"model_max_length": 1000000000000000019884624838656,
|
||||
"tokenizer_class": "GPT2Tokenizer",
|
||||
"unk_token": "<|endoftext|>",
|
||||
"vocab_size": 50304
|
||||
}
|
||||
Reference in New Issue
Block a user