Text Generation
Transformers
Safetensors
afmoe
reasoning
agentic
tool-calling
thinking
conversational
custom_code
compressed-tensors
Instructions to use arcee-ai/Trinity-Large-Thinking-W4A16 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use arcee-ai/Trinity-Large-Thinking-W4A16 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="arcee-ai/Trinity-Large-Thinking-W4A16", trust_remote_code=True) messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("arcee-ai/Trinity-Large-Thinking-W4A16", trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained("arcee-ai/Trinity-Large-Thinking-W4A16", trust_remote_code=True, device_map="auto") messages = [ {"role": "user", "content": "Who are you?"}, ] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use arcee-ai/Trinity-Large-Thinking-W4A16 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "arcee-ai/Trinity-Large-Thinking-W4A16" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "arcee-ai/Trinity-Large-Thinking-W4A16", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/arcee-ai/Trinity-Large-Thinking-W4A16
- SGLang
How to use arcee-ai/Trinity-Large-Thinking-W4A16 with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "arcee-ai/Trinity-Large-Thinking-W4A16" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "arcee-ai/Trinity-Large-Thinking-W4A16", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "arcee-ai/Trinity-Large-Thinking-W4A16" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "arcee-ai/Trinity-Large-Thinking-W4A16", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use arcee-ai/Trinity-Large-Thinking-W4A16 with Docker Model Runner:
docker model run hf.co/arcee-ai/Trinity-Large-Thinking-W4A16
Anneketh Vij commited on
Commit ·
45ce26d
0
Parent(s):
Super-squash branch 'main' using huggingface_hub
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- .gitattributes +37 -0
- README.md +122 -0
- __init__.py +4 -0
- chat_template.jinja +159 -0
- config.json +499 -0
- configuration_afmoe.py +133 -0
- generation_config.json +7 -0
- model-00001-of-00043.safetensors +3 -0
- model-00002-of-00043.safetensors +3 -0
- model-00003-of-00043.safetensors +3 -0
- model-00004-of-00043.safetensors +3 -0
- model-00005-of-00043.safetensors +3 -0
- model-00006-of-00043.safetensors +3 -0
- model-00007-of-00043.safetensors +3 -0
- model-00008-of-00043.safetensors +3 -0
- model-00009-of-00043.safetensors +3 -0
- model-00010-of-00043.safetensors +3 -0
- model-00011-of-00043.safetensors +3 -0
- model-00012-of-00043.safetensors +3 -0
- model-00013-of-00043.safetensors +3 -0
- model-00014-of-00043.safetensors +3 -0
- model-00015-of-00043.safetensors +3 -0
- model-00016-of-00043.safetensors +3 -0
- model-00017-of-00043.safetensors +3 -0
- model-00018-of-00043.safetensors +3 -0
- model-00019-of-00043.safetensors +3 -0
- model-00020-of-00043.safetensors +3 -0
- model-00021-of-00043.safetensors +3 -0
- model-00022-of-00043.safetensors +3 -0
- model-00023-of-00043.safetensors +3 -0
- model-00024-of-00043.safetensors +3 -0
- model-00025-of-00043.safetensors +3 -0
- model-00026-of-00043.safetensors +3 -0
- model-00027-of-00043.safetensors +3 -0
- model-00028-of-00043.safetensors +3 -0
- model-00029-of-00043.safetensors +3 -0
- model-00030-of-00043.safetensors +3 -0
- model-00031-of-00043.safetensors +3 -0
- model-00032-of-00043.safetensors +3 -0
- model-00033-of-00043.safetensors +3 -0
- model-00034-of-00043.safetensors +3 -0
- model-00035-of-00043.safetensors +3 -0
- model-00036-of-00043.safetensors +3 -0
- model-00037-of-00043.safetensors +3 -0
- model-00038-of-00043.safetensors +3 -0
- model-00039-of-00043.safetensors +3 -0
- model-00040-of-00043.safetensors +3 -0
- model-00041-of-00043.safetensors +3 -0
- model-00042-of-00043.safetensors +3 -0
- model-00043-of-00043.safetensors +3 -0
.gitattributes
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
model.safetensors.index.json filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,122 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
- es
|
| 6 |
+
- fr
|
| 7 |
+
- de
|
| 8 |
+
- it
|
| 9 |
+
- pt
|
| 10 |
+
- ru
|
| 11 |
+
- ar
|
| 12 |
+
- hi
|
| 13 |
+
- ko
|
| 14 |
+
- zh
|
| 15 |
+
library_name: transformers
|
| 16 |
+
base_model:
|
| 17 |
+
- arcee-ai/Trinity-Large-Thinking
|
| 18 |
+
base_model_relation: quantized
|
| 19 |
+
tags:
|
| 20 |
+
- reasoning
|
| 21 |
+
- agentic
|
| 22 |
+
- tool-calling
|
| 23 |
+
- thinking
|
| 24 |
+
---
|
| 25 |
+
<!-- markdownlint-disable first-line-h1 -->
|
| 26 |
+
<!-- markdownlint-disable html -->
|
| 27 |
+
<!-- markdownlint-disable no-duplicate-header -->
|
| 28 |
+
|
| 29 |
+
<div align="center">
|
| 30 |
+
<picture>
|
| 31 |
+
<img
|
| 32 |
+
src="https://cdn-uploads.huggingface.co/production/uploads/6435718aaaef013d1aec3b8b/i-v1KyAMOW_mgVGeic9WJ.png"
|
| 33 |
+
alt="Arcee Trinity Large Thinking"
|
| 34 |
+
style="max-width: 100%; height: auto;"
|
| 35 |
+
>
|
| 36 |
+
</picture>
|
| 37 |
+
</div>
|
| 38 |
+
<hr>
|
| 39 |
+
|
| 40 |
+
# Trinity-Large-Thinking-W4A16
|
| 41 |
+
|
| 42 |
+
## Introduction
|
| 43 |
+
|
| 44 |
+
Trinity-Large-Thinking is a reasoning-optimized variant of Arcee AI's Trinity-Large family — a 398B-parameter sparse Mixture-of-Experts (MoE) model with approximately 13B active parameters per token, post-trained with extended chain-of-thought reasoning and agentic RL.
|
| 45 |
+
|
| 46 |
+
**This repository contains the W4A16 quantized weights of Trinity-Large-Thinking (INT4 weights, 16-bit activations).**
|
| 47 |
+
|
| 48 |
+
For full model details, benchmarks, and usage guidance, see the main [Trinity-Large-Thinking](https://huggingface.co/arcee-ai/Trinity-Large-Thinking) model card.
|
| 49 |
+
|
| 50 |
+
## Quantization Details
|
| 51 |
+
|
| 52 |
+
- **Scheme:** `W4A16` (INT4 weights, 16-bit activations)
|
| 53 |
+
- **Intended use:** Quality-preserving 4-bit deployment of Trinity-Large-Thinking
|
| 54 |
+
|
| 55 |
+
## Usage
|
| 56 |
+
|
| 57 |
+
### Inference tested on
|
| 58 |
+
|
| 59 |
+
- 8x NVIDIA H100 80GB (tensor parallel = 8)
|
| 60 |
+
- vLLM 0.15.1+
|
| 61 |
+
|
| 62 |
+
### vLLM
|
| 63 |
+
|
| 64 |
+
Supported in vLLM 0.15.1+.
|
| 65 |
+
|
| 66 |
+
```bash
|
| 67 |
+
vllm serve arcee-ai/Trinity-Large-Thinking-W4A16 \
|
| 68 |
+
--trust-remote-code \
|
| 69 |
+
--tensor-parallel-size 8 \
|
| 70 |
+
--enable-reasoning \
|
| 71 |
+
--reasoning-parser deepseek_r1 \
|
| 72 |
+
--enable-auto-tool-choice \
|
| 73 |
+
--tool-call-parser qwen3_coder
|
| 74 |
+
```
|
| 75 |
+
|
| 76 |
+
### Transformers
|
| 77 |
+
|
| 78 |
+
```python
|
| 79 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 80 |
+
|
| 81 |
+
model_id = "arcee-ai/Trinity-Large-Thinking-W4A16"
|
| 82 |
+
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 83 |
+
model = AutoModelForCausalLM.from_pretrained(
|
| 84 |
+
model_id,
|
| 85 |
+
device_map="auto",
|
| 86 |
+
trust_remote_code=True
|
| 87 |
+
)
|
| 88 |
+
|
| 89 |
+
messages = [{"role": "user", "content": "Who are you?"}]
|
| 90 |
+
|
| 91 |
+
input_ids = tokenizer.apply_chat_template(
|
| 92 |
+
messages, add_generation_prompt=True, return_tensors="pt"
|
| 93 |
+
).to(model.device)
|
| 94 |
+
|
| 95 |
+
outputs = model.generate(input_ids, max_new_tokens=4096, do_sample=True, temperature=0.6, top_k=50, top_p=0.95)
|
| 96 |
+
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
| 97 |
+
```
|
| 98 |
+
|
| 99 |
+
### API
|
| 100 |
+
|
| 101 |
+
Works out of the box on [OpenRouter](https://openrouter.ai/) as `arcee-ai/trinity-large-thinking`.
|
| 102 |
+
|
| 103 |
+
## License
|
| 104 |
+
|
| 105 |
+
Trinity-Large-Thinking-W4A16 is released under the Apache License, Version 2.0.
|
| 106 |
+
|
| 107 |
+
## Citation
|
| 108 |
+
|
| 109 |
+
If you use this model, please cite:
|
| 110 |
+
|
| 111 |
+
```bibtex
|
| 112 |
+
@misc{singh2026arceetrinity,
|
| 113 |
+
title = {Arcee Trinity Large Technical Report},
|
| 114 |
+
author = {Varun Singh and Lucas Krauss and Sami Jaghouar and Matej Sirovatka and Charles Goddard and Fares Obied and Jack Min Ong and Jannik Straube and Fern and Aria Harley and Conner Stewart and Colin Kealty and Maziyar Panahi and Simon Kirsten and Anushka Deshpande and Anneketh Vij and Arthur Bresnu and Pranav Veldurthi and Raghav Ravishankar and Hardik Bishnoi and DatologyAI Team and Arcee AI Team and Prime Intellect Team and Mark McQuade and Johannes Hagemann and Lucas Atkins},
|
| 115 |
+
year = {2026},
|
| 116 |
+
eprint = {2602.17004},
|
| 117 |
+
archivePrefix= {arXiv},
|
| 118 |
+
primaryClass = {cs.LG},
|
| 119 |
+
doi = {10.48550/arXiv.2602.17004},
|
| 120 |
+
url = {https://arxiv.org/abs/2602.17004}
|
| 121 |
+
}
|
| 122 |
+
```
|
__init__.py
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from .configuration_afmoe import AfmoeConfig
|
| 2 |
+
from .modeling_afmoe import AfmoeForCausalLM
|
| 3 |
+
|
| 4 |
+
__all__ = ["AfmoeConfig", "AfmoeForCausalLM"]
|
chat_template.jinja
ADDED
|
@@ -0,0 +1,159 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<|begin_of_text|>{%- macro render_extra_keys(json_dict, handled_keys) -%}
|
| 2 |
+
{%- if json_dict is mapping %}
|
| 3 |
+
{%- for json_key in json_dict if json_key not in handled_keys %}
|
| 4 |
+
{%- if json_dict[json_key] is mapping or (json_dict[json_key] is sequence and json_dict[json_key] is not string) %}
|
| 5 |
+
{{- '\n<' ~ json_key ~ '>' ~ (json_dict[json_key] | tojson | safe) ~ '</' ~ json_key ~ '>' }}
|
| 6 |
+
{%- else %}
|
| 7 |
+
{{- '\n<' ~ json_key ~ '>' ~ (json_dict[json_key] | string) ~ '</' ~ json_key ~ '>' }}
|
| 8 |
+
{%- endif %}
|
| 9 |
+
{%- endfor %}
|
| 10 |
+
{%- endif %}
|
| 11 |
+
{%- endmacro -%}
|
| 12 |
+
|
| 13 |
+
{%- macro render_tool_call(raw_tool_call) -%}
|
| 14 |
+
{%- if raw_tool_call.function is defined and raw_tool_call.function is mapping %}
|
| 15 |
+
{%- set tool_call = raw_tool_call.function %}
|
| 16 |
+
{%- else %}
|
| 17 |
+
{%- set tool_call = raw_tool_call %}
|
| 18 |
+
{%- endif %}
|
| 19 |
+
{{- '<tool_call>\n<function=' + (tool_call.name | default('') | string) + '>\n' }}
|
| 20 |
+
{%- if tool_call.arguments is defined and tool_call.arguments is mapping %}
|
| 21 |
+
{%- for args_name, args_value in tool_call.arguments.items() %}
|
| 22 |
+
{{- '<parameter=' + (args_name | string) + '>\n' }}
|
| 23 |
+
{%- if args_value is mapping or (args_value is sequence and args_value is not string) %}
|
| 24 |
+
{{- args_value | tojson | safe }}
|
| 25 |
+
{%- else %}
|
| 26 |
+
{{- args_value | string }}
|
| 27 |
+
{%- endif %}
|
| 28 |
+
{{- '\n</parameter>\n' }}
|
| 29 |
+
{%- endfor %}
|
| 30 |
+
{%- endif %}
|
| 31 |
+
{{- '</function>\n</tool_call>' }}
|
| 32 |
+
{%- endmacro -%}
|
| 33 |
+
|
| 34 |
+
{%- set system_message = none %}
|
| 35 |
+
{%- if messages and messages[0]["role"] == "system" %}
|
| 36 |
+
{%- set system_message = messages[0]["content"] %}
|
| 37 |
+
{%- set loop_messages = messages[1:] %}
|
| 38 |
+
{%- else %}
|
| 39 |
+
{%- set loop_messages = messages %}
|
| 40 |
+
{%- endif %}
|
| 41 |
+
|
| 42 |
+
{%- if not tools is defined %}
|
| 43 |
+
{%- set tools = [] %}
|
| 44 |
+
{%- endif %}
|
| 45 |
+
{%- set has_tools = tools is iterable and tools is not string and tools | length > 0 %}
|
| 46 |
+
|
| 47 |
+
{%- if system_message is not none or has_tools %}
|
| 48 |
+
{{- '<|im_start|>system\n' }}
|
| 49 |
+
{%- if system_message is not none %}
|
| 50 |
+
{{- system_message }}
|
| 51 |
+
{%- else %}
|
| 52 |
+
{{- "You are Trinity Large, a helpful assistant developed by Arcee AI, that can interact with a computer to solve tasks." }}
|
| 53 |
+
{%- endif %}
|
| 54 |
+
{%- if has_tools %}
|
| 55 |
+
{{- "\n\n# Tools\n\nYou have access to the following functions:\n\n<tools>" }}
|
| 56 |
+
{%- for tool in tools %}
|
| 57 |
+
{%- if tool.function is defined and tool.function is mapping %}
|
| 58 |
+
{%- set tool = tool.function %}
|
| 59 |
+
{%- endif %}
|
| 60 |
+
{{- '\n<function>\n<name>' ~ (tool.name | default('') | string) ~ '</name>' }}
|
| 61 |
+
{%- if tool.description is defined and tool.description is not none %}
|
| 62 |
+
{{- '\n<description>' ~ (tool.description | string | trim) ~ '</description>' }}
|
| 63 |
+
{%- endif %}
|
| 64 |
+
{{- '\n<parameters>' }}
|
| 65 |
+
{%- if tool.parameters is defined and tool.parameters is mapping and tool.parameters.properties is defined and tool.parameters.properties is mapping %}
|
| 66 |
+
{%- for param_name, param_fields in tool.parameters.properties.items() %}
|
| 67 |
+
{{- '\n<parameter>\n<name>' ~ (param_name | string) ~ '</name>' }}
|
| 68 |
+
{%- if param_fields is mapping and param_fields.type is defined and param_fields.type is not none %}
|
| 69 |
+
{{- '\n<type>' ~ (param_fields.type | string) ~ '</type>' }}
|
| 70 |
+
{%- endif %}
|
| 71 |
+
{%- if param_fields is mapping and param_fields.description is defined and param_fields.description is not none %}
|
| 72 |
+
{{- '\n<description>' ~ (param_fields.description | string | trim) ~ '</description>' }}
|
| 73 |
+
{%- endif %}
|
| 74 |
+
{%- if param_fields is mapping %}
|
| 75 |
+
{%- set handled_keys = ['name', 'type', 'description'] %}
|
| 76 |
+
{{- render_extra_keys(param_fields, handled_keys) }}
|
| 77 |
+
{%- endif %}
|
| 78 |
+
{{- '\n</parameter>' }}
|
| 79 |
+
{%- endfor %}
|
| 80 |
+
{%- endif %}
|
| 81 |
+
{%- if tool.parameters is defined %}
|
| 82 |
+
{%- set handled_keys = ['type', 'properties'] %}
|
| 83 |
+
{{- render_extra_keys(tool.parameters, handled_keys) }}
|
| 84 |
+
{%- endif %}
|
| 85 |
+
{{- '\n</parameters>' }}
|
| 86 |
+
{%- set handled_keys = ['type', 'name', 'description', 'parameters'] %}
|
| 87 |
+
{{- render_extra_keys(tool, handled_keys) }}
|
| 88 |
+
{{- '\n</function>' }}
|
| 89 |
+
{%- endfor %}
|
| 90 |
+
{{- "\n</tools>" }}
|
| 91 |
+
{{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n<tool_call>\n<function=example_function_name>\n<parameter=example_parameter_1>\nvalue_1\n</parameter>\n<parameter=example_parameter_2>\nThis is the value for the second parameter\nthat can span\nmultiple lines\n</parameter>\n</function>\n</tool_call>\n\n<IMPORTANT>\nReminder:\n- Function calls MUST follow the specified format: an inner <function=...></function> block must be nested within <tool_call></tool_call> XML tags\n- Required parameters MUST be specified\n- You may provide optional reasoning for your function call in natural language BEFORE the function call, but NOT after\n- If there is no function call available, answer the question like normal with your current knowledge and do not tell the user about function calls\n</IMPORTANT>' }}
|
| 92 |
+
{%- endif %}
|
| 93 |
+
{{- '<|im_end|>\n' }}
|
| 94 |
+
{%- endif %}
|
| 95 |
+
|
| 96 |
+
{%- for message in loop_messages %}
|
| 97 |
+
{%- set role = message.role | default('') %}
|
| 98 |
+
{%- if role == "assistant" %}
|
| 99 |
+
{%- set content_str = '' if message.content is none else (message.content | string) %}
|
| 100 |
+
{%- set trimmed_content = content_str | trim %}
|
| 101 |
+
|
| 102 |
+
{%- set has_reasoning_content = message.reasoning_content is defined %}
|
| 103 |
+
{%- set has_reasoning = has_reasoning_content or (message.reasoning is defined) %}
|
| 104 |
+
|
| 105 |
+
{%- if has_reasoning_content %}
|
| 106 |
+
{%- set reasoning_value = message.reasoning_content %}
|
| 107 |
+
{%- elif message.reasoning is defined %}
|
| 108 |
+
{%- set reasoning_value = message.reasoning %}
|
| 109 |
+
{%- else %}
|
| 110 |
+
{%- set reasoning_value = none %}
|
| 111 |
+
{%- endif %}
|
| 112 |
+
|
| 113 |
+
{%- set has_tool_calls = message.tool_calls is defined and message.tool_calls is iterable and message.tool_calls is not string and message.tool_calls | length > 0 %}
|
| 114 |
+
|
| 115 |
+
{{- '<|im_start|>assistant\n' }}
|
| 116 |
+
{%- if has_reasoning %}
|
| 117 |
+
{%- if reasoning_value %}
|
| 118 |
+
{{- '<think>' + (reasoning_value | string | trim) + '</think>' }}
|
| 119 |
+
{%- else %}
|
| 120 |
+
{{- '<think></think>' }}
|
| 121 |
+
{%- endif %}
|
| 122 |
+
{%- if trimmed_content %}
|
| 123 |
+
{{- '\n' + trimmed_content }}
|
| 124 |
+
{%- endif %}
|
| 125 |
+
{%- elif has_tool_calls %}
|
| 126 |
+
{%- if trimmed_content %}
|
| 127 |
+
{{- trimmed_content }}
|
| 128 |
+
{%- endif %}
|
| 129 |
+
{%- else %}
|
| 130 |
+
{{- content_str }}
|
| 131 |
+
{%- endif %}
|
| 132 |
+
|
| 133 |
+
{%- if has_tool_calls %}
|
| 134 |
+
{%- for tool_call in message.tool_calls %}
|
| 135 |
+
{%- set separator = '\n' if ((loop.first and (has_reasoning or trimmed_content)) or (not loop.first)) else '' -%}
|
| 136 |
+
{{- separator + render_tool_call(tool_call) }}
|
| 137 |
+
{%- endfor %}
|
| 138 |
+
{%- endif %}
|
| 139 |
+
{{- '<|im_end|>\n' }}
|
| 140 |
+
{%- elif role == "tool" or role == "observation" or role == "function" %}
|
| 141 |
+
{%- if loop.first or loop.previtem.role not in ["tool", "observation", "function"] %}
|
| 142 |
+
{{- '<|im_start|>user\n' }}
|
| 143 |
+
{%- endif %}
|
| 144 |
+
{{- '<tool_response>\n' }}
|
| 145 |
+
{{- '' if message.content is none else (message.content | string) }}
|
| 146 |
+
{{- '\n</tool_response>\n' }}
|
| 147 |
+
{%- if loop.last or loop.nextitem.role not in ["tool", "observation", "function"] %}
|
| 148 |
+
{{- '<|im_end|>\n' }}
|
| 149 |
+
{%- endif %}
|
| 150 |
+
{%- else %}
|
| 151 |
+
{{- '<|im_start|>' + (role | string) }}
|
| 152 |
+
{{- '\n' + ('' if message.content is none else (message.content | string)) }}
|
| 153 |
+
{{- '<|im_end|>\n' }}
|
| 154 |
+
{%- endif %}
|
| 155 |
+
{%- endfor %}
|
| 156 |
+
|
| 157 |
+
{%- if add_generation_prompt %}
|
| 158 |
+
{{- '<|im_start|>assistant\n<think>' }}
|
| 159 |
+
{%- endif %}
|
config.json
ADDED
|
@@ -0,0 +1,499 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"AfmoeForCausalLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_dropout": 0.0,
|
| 6 |
+
"auto_map": {
|
| 7 |
+
"AutoConfig": "configuration_afmoe.AfmoeConfig",
|
| 8 |
+
"AutoModel": "modeling_afmoe.AfmoeModel",
|
| 9 |
+
"AutoModelForCausalLM": "modeling_afmoe.AfmoeForCausalLM"
|
| 10 |
+
},
|
| 11 |
+
"dtype": "bfloat16",
|
| 12 |
+
"global_attn_every_n_layers": 4,
|
| 13 |
+
"head_dim": 128,
|
| 14 |
+
"hidden_act": "silu",
|
| 15 |
+
"hidden_size": 3072,
|
| 16 |
+
"initializer_range": 0.02,
|
| 17 |
+
"intermediate_size": 12288,
|
| 18 |
+
"layer_types": [
|
| 19 |
+
"sliding_attention",
|
| 20 |
+
"sliding_attention",
|
| 21 |
+
"sliding_attention",
|
| 22 |
+
"full_attention",
|
| 23 |
+
"sliding_attention",
|
| 24 |
+
"sliding_attention",
|
| 25 |
+
"sliding_attention",
|
| 26 |
+
"full_attention",
|
| 27 |
+
"sliding_attention",
|
| 28 |
+
"sliding_attention",
|
| 29 |
+
"sliding_attention",
|
| 30 |
+
"full_attention",
|
| 31 |
+
"sliding_attention",
|
| 32 |
+
"sliding_attention",
|
| 33 |
+
"sliding_attention",
|
| 34 |
+
"full_attention",
|
| 35 |
+
"sliding_attention",
|
| 36 |
+
"sliding_attention",
|
| 37 |
+
"sliding_attention",
|
| 38 |
+
"full_attention",
|
| 39 |
+
"sliding_attention",
|
| 40 |
+
"sliding_attention",
|
| 41 |
+
"sliding_attention",
|
| 42 |
+
"full_attention",
|
| 43 |
+
"sliding_attention",
|
| 44 |
+
"sliding_attention",
|
| 45 |
+
"sliding_attention",
|
| 46 |
+
"full_attention",
|
| 47 |
+
"sliding_attention",
|
| 48 |
+
"sliding_attention",
|
| 49 |
+
"sliding_attention",
|
| 50 |
+
"full_attention",
|
| 51 |
+
"sliding_attention",
|
| 52 |
+
"sliding_attention",
|
| 53 |
+
"sliding_attention",
|
| 54 |
+
"full_attention",
|
| 55 |
+
"sliding_attention",
|
| 56 |
+
"sliding_attention",
|
| 57 |
+
"sliding_attention",
|
| 58 |
+
"full_attention",
|
| 59 |
+
"sliding_attention",
|
| 60 |
+
"sliding_attention",
|
| 61 |
+
"sliding_attention",
|
| 62 |
+
"full_attention",
|
| 63 |
+
"sliding_attention",
|
| 64 |
+
"sliding_attention",
|
| 65 |
+
"sliding_attention",
|
| 66 |
+
"full_attention",
|
| 67 |
+
"sliding_attention",
|
| 68 |
+
"sliding_attention",
|
| 69 |
+
"sliding_attention",
|
| 70 |
+
"full_attention",
|
| 71 |
+
"sliding_attention",
|
| 72 |
+
"sliding_attention",
|
| 73 |
+
"sliding_attention",
|
| 74 |
+
"full_attention",
|
| 75 |
+
"sliding_attention",
|
| 76 |
+
"sliding_attention",
|
| 77 |
+
"sliding_attention",
|
| 78 |
+
"full_attention"
|
| 79 |
+
],
|
| 80 |
+
"load_balance_coeff": 5e-05,
|
| 81 |
+
"max_position_embeddings": 262144,
|
| 82 |
+
"model_type": "afmoe",
|
| 83 |
+
"moe_intermediate_size": 3072,
|
| 84 |
+
"mup_enabled": true,
|
| 85 |
+
"n_group": 1,
|
| 86 |
+
"num_attention_heads": 48,
|
| 87 |
+
"num_dense_layers": 6,
|
| 88 |
+
"num_expert_groups": 1,
|
| 89 |
+
"num_experts": 256,
|
| 90 |
+
"num_experts_per_tok": 4,
|
| 91 |
+
"num_hidden_layers": 60,
|
| 92 |
+
"num_key_value_heads": 8,
|
| 93 |
+
"num_limited_groups": 1,
|
| 94 |
+
"num_shared_experts": 1,
|
| 95 |
+
"quantization_config": {
|
| 96 |
+
"config_groups": {
|
| 97 |
+
"group_0": {
|
| 98 |
+
"format": "pack-quantized",
|
| 99 |
+
"input_activations": null,
|
| 100 |
+
"output_activations": null,
|
| 101 |
+
"targets": [
|
| 102 |
+
"Linear"
|
| 103 |
+
],
|
| 104 |
+
"weights": {
|
| 105 |
+
"actorder": null,
|
| 106 |
+
"block_structure": null,
|
| 107 |
+
"dynamic": false,
|
| 108 |
+
"group_size": 128,
|
| 109 |
+
"num_bits": 4,
|
| 110 |
+
"observer": "memoryless_minmax",
|
| 111 |
+
"observer_kwargs": {},
|
| 112 |
+
"scale_dtype": null,
|
| 113 |
+
"strategy": "group",
|
| 114 |
+
"symmetric": true,
|
| 115 |
+
"type": "int",
|
| 116 |
+
"zp_dtype": null
|
| 117 |
+
}
|
| 118 |
+
}
|
| 119 |
+
},
|
| 120 |
+
"format": "pack-quantized",
|
| 121 |
+
"global_compression_ratio": null,
|
| 122 |
+
"ignore": [
|
| 123 |
+
"model.layers.0.self_attn.q_proj",
|
| 124 |
+
"model.layers.0.self_attn.k_proj",
|
| 125 |
+
"model.layers.0.self_attn.v_proj",
|
| 126 |
+
"model.layers.0.self_attn.o_proj",
|
| 127 |
+
"model.layers.0.self_attn.gate_proj",
|
| 128 |
+
"model.layers.1.self_attn.q_proj",
|
| 129 |
+
"model.layers.1.self_attn.k_proj",
|
| 130 |
+
"model.layers.1.self_attn.v_proj",
|
| 131 |
+
"model.layers.1.self_attn.o_proj",
|
| 132 |
+
"model.layers.1.self_attn.gate_proj",
|
| 133 |
+
"model.layers.2.self_attn.q_proj",
|
| 134 |
+
"model.layers.2.self_attn.k_proj",
|
| 135 |
+
"model.layers.2.self_attn.v_proj",
|
| 136 |
+
"model.layers.2.self_attn.o_proj",
|
| 137 |
+
"model.layers.2.self_attn.gate_proj",
|
| 138 |
+
"model.layers.3.self_attn.q_proj",
|
| 139 |
+
"model.layers.3.self_attn.k_proj",
|
| 140 |
+
"model.layers.3.self_attn.v_proj",
|
| 141 |
+
"model.layers.3.self_attn.o_proj",
|
| 142 |
+
"model.layers.3.self_attn.gate_proj",
|
| 143 |
+
"model.layers.4.self_attn.q_proj",
|
| 144 |
+
"model.layers.4.self_attn.k_proj",
|
| 145 |
+
"model.layers.4.self_attn.v_proj",
|
| 146 |
+
"model.layers.4.self_attn.o_proj",
|
| 147 |
+
"model.layers.4.self_attn.gate_proj",
|
| 148 |
+
"model.layers.5.self_attn.q_proj",
|
| 149 |
+
"model.layers.5.self_attn.k_proj",
|
| 150 |
+
"model.layers.5.self_attn.v_proj",
|
| 151 |
+
"model.layers.5.self_attn.o_proj",
|
| 152 |
+
"model.layers.5.self_attn.gate_proj",
|
| 153 |
+
"model.layers.6.self_attn.q_proj",
|
| 154 |
+
"model.layers.6.self_attn.k_proj",
|
| 155 |
+
"model.layers.6.self_attn.v_proj",
|
| 156 |
+
"model.layers.6.self_attn.o_proj",
|
| 157 |
+
"model.layers.6.self_attn.gate_proj",
|
| 158 |
+
"model.layers.6.mlp.router.gate",
|
| 159 |
+
"model.layers.7.self_attn.q_proj",
|
| 160 |
+
"model.layers.7.self_attn.k_proj",
|
| 161 |
+
"model.layers.7.self_attn.v_proj",
|
| 162 |
+
"model.layers.7.self_attn.o_proj",
|
| 163 |
+
"model.layers.7.self_attn.gate_proj",
|
| 164 |
+
"model.layers.7.mlp.router.gate",
|
| 165 |
+
"model.layers.8.self_attn.q_proj",
|
| 166 |
+
"model.layers.8.self_attn.k_proj",
|
| 167 |
+
"model.layers.8.self_attn.v_proj",
|
| 168 |
+
"model.layers.8.self_attn.o_proj",
|
| 169 |
+
"model.layers.8.self_attn.gate_proj",
|
| 170 |
+
"model.layers.8.mlp.router.gate",
|
| 171 |
+
"model.layers.9.self_attn.q_proj",
|
| 172 |
+
"model.layers.9.self_attn.k_proj",
|
| 173 |
+
"model.layers.9.self_attn.v_proj",
|
| 174 |
+
"model.layers.9.self_attn.o_proj",
|
| 175 |
+
"model.layers.9.self_attn.gate_proj",
|
| 176 |
+
"model.layers.9.mlp.router.gate",
|
| 177 |
+
"model.layers.10.self_attn.q_proj",
|
| 178 |
+
"model.layers.10.self_attn.k_proj",
|
| 179 |
+
"model.layers.10.self_attn.v_proj",
|
| 180 |
+
"model.layers.10.self_attn.o_proj",
|
| 181 |
+
"model.layers.10.self_attn.gate_proj",
|
| 182 |
+
"model.layers.10.mlp.router.gate",
|
| 183 |
+
"model.layers.11.self_attn.q_proj",
|
| 184 |
+
"model.layers.11.self_attn.k_proj",
|
| 185 |
+
"model.layers.11.self_attn.v_proj",
|
| 186 |
+
"model.layers.11.self_attn.o_proj",
|
| 187 |
+
"model.layers.11.self_attn.gate_proj",
|
| 188 |
+
"model.layers.11.mlp.router.gate",
|
| 189 |
+
"model.layers.12.self_attn.q_proj",
|
| 190 |
+
"model.layers.12.self_attn.k_proj",
|
| 191 |
+
"model.layers.12.self_attn.v_proj",
|
| 192 |
+
"model.layers.12.self_attn.o_proj",
|
| 193 |
+
"model.layers.12.self_attn.gate_proj",
|
| 194 |
+
"model.layers.12.mlp.router.gate",
|
| 195 |
+
"model.layers.13.self_attn.q_proj",
|
| 196 |
+
"model.layers.13.self_attn.k_proj",
|
| 197 |
+
"model.layers.13.self_attn.v_proj",
|
| 198 |
+
"model.layers.13.self_attn.o_proj",
|
| 199 |
+
"model.layers.13.self_attn.gate_proj",
|
| 200 |
+
"model.layers.13.mlp.router.gate",
|
| 201 |
+
"model.layers.14.self_attn.q_proj",
|
| 202 |
+
"model.layers.14.self_attn.k_proj",
|
| 203 |
+
"model.layers.14.self_attn.v_proj",
|
| 204 |
+
"model.layers.14.self_attn.o_proj",
|
| 205 |
+
"model.layers.14.self_attn.gate_proj",
|
| 206 |
+
"model.layers.14.mlp.router.gate",
|
| 207 |
+
"model.layers.15.self_attn.q_proj",
|
| 208 |
+
"model.layers.15.self_attn.k_proj",
|
| 209 |
+
"model.layers.15.self_attn.v_proj",
|
| 210 |
+
"model.layers.15.self_attn.o_proj",
|
| 211 |
+
"model.layers.15.self_attn.gate_proj",
|
| 212 |
+
"model.layers.15.mlp.router.gate",
|
| 213 |
+
"model.layers.16.self_attn.q_proj",
|
| 214 |
+
"model.layers.16.self_attn.k_proj",
|
| 215 |
+
"model.layers.16.self_attn.v_proj",
|
| 216 |
+
"model.layers.16.self_attn.o_proj",
|
| 217 |
+
"model.layers.16.self_attn.gate_proj",
|
| 218 |
+
"model.layers.16.mlp.router.gate",
|
| 219 |
+
"model.layers.17.self_attn.q_proj",
|
| 220 |
+
"model.layers.17.self_attn.k_proj",
|
| 221 |
+
"model.layers.17.self_attn.v_proj",
|
| 222 |
+
"model.layers.17.self_attn.o_proj",
|
| 223 |
+
"model.layers.17.self_attn.gate_proj",
|
| 224 |
+
"model.layers.17.mlp.router.gate",
|
| 225 |
+
"model.layers.18.self_attn.q_proj",
|
| 226 |
+
"model.layers.18.self_attn.k_proj",
|
| 227 |
+
"model.layers.18.self_attn.v_proj",
|
| 228 |
+
"model.layers.18.self_attn.o_proj",
|
| 229 |
+
"model.layers.18.self_attn.gate_proj",
|
| 230 |
+
"model.layers.18.mlp.router.gate",
|
| 231 |
+
"model.layers.19.self_attn.q_proj",
|
| 232 |
+
"model.layers.19.self_attn.k_proj",
|
| 233 |
+
"model.layers.19.self_attn.v_proj",
|
| 234 |
+
"model.layers.19.self_attn.o_proj",
|
| 235 |
+
"model.layers.19.self_attn.gate_proj",
|
| 236 |
+
"model.layers.19.mlp.router.gate",
|
| 237 |
+
"model.layers.20.self_attn.q_proj",
|
| 238 |
+
"model.layers.20.self_attn.k_proj",
|
| 239 |
+
"model.layers.20.self_attn.v_proj",
|
| 240 |
+
"model.layers.20.self_attn.o_proj",
|
| 241 |
+
"model.layers.20.self_attn.gate_proj",
|
| 242 |
+
"model.layers.20.mlp.router.gate",
|
| 243 |
+
"model.layers.21.self_attn.q_proj",
|
| 244 |
+
"model.layers.21.self_attn.k_proj",
|
| 245 |
+
"model.layers.21.self_attn.v_proj",
|
| 246 |
+
"model.layers.21.self_attn.o_proj",
|
| 247 |
+
"model.layers.21.self_attn.gate_proj",
|
| 248 |
+
"model.layers.21.mlp.router.gate",
|
| 249 |
+
"model.layers.22.self_attn.q_proj",
|
| 250 |
+
"model.layers.22.self_attn.k_proj",
|
| 251 |
+
"model.layers.22.self_attn.v_proj",
|
| 252 |
+
"model.layers.22.self_attn.o_proj",
|
| 253 |
+
"model.layers.22.self_attn.gate_proj",
|
| 254 |
+
"model.layers.22.mlp.router.gate",
|
| 255 |
+
"model.layers.23.self_attn.q_proj",
|
| 256 |
+
"model.layers.23.self_attn.k_proj",
|
| 257 |
+
"model.layers.23.self_attn.v_proj",
|
| 258 |
+
"model.layers.23.self_attn.o_proj",
|
| 259 |
+
"model.layers.23.self_attn.gate_proj",
|
| 260 |
+
"model.layers.23.mlp.router.gate",
|
| 261 |
+
"model.layers.24.self_attn.q_proj",
|
| 262 |
+
"model.layers.24.self_attn.k_proj",
|
| 263 |
+
"model.layers.24.self_attn.v_proj",
|
| 264 |
+
"model.layers.24.self_attn.o_proj",
|
| 265 |
+
"model.layers.24.self_attn.gate_proj",
|
| 266 |
+
"model.layers.24.mlp.router.gate",
|
| 267 |
+
"model.layers.25.self_attn.q_proj",
|
| 268 |
+
"model.layers.25.self_attn.k_proj",
|
| 269 |
+
"model.layers.25.self_attn.v_proj",
|
| 270 |
+
"model.layers.25.self_attn.o_proj",
|
| 271 |
+
"model.layers.25.self_attn.gate_proj",
|
| 272 |
+
"model.layers.25.mlp.router.gate",
|
| 273 |
+
"model.layers.26.self_attn.q_proj",
|
| 274 |
+
"model.layers.26.self_attn.k_proj",
|
| 275 |
+
"model.layers.26.self_attn.v_proj",
|
| 276 |
+
"model.layers.26.self_attn.o_proj",
|
| 277 |
+
"model.layers.26.self_attn.gate_proj",
|
| 278 |
+
"model.layers.26.mlp.router.gate",
|
| 279 |
+
"model.layers.27.self_attn.q_proj",
|
| 280 |
+
"model.layers.27.self_attn.k_proj",
|
| 281 |
+
"model.layers.27.self_attn.v_proj",
|
| 282 |
+
"model.layers.27.self_attn.o_proj",
|
| 283 |
+
"model.layers.27.self_attn.gate_proj",
|
| 284 |
+
"model.layers.27.mlp.router.gate",
|
| 285 |
+
"model.layers.28.self_attn.q_proj",
|
| 286 |
+
"model.layers.28.self_attn.k_proj",
|
| 287 |
+
"model.layers.28.self_attn.v_proj",
|
| 288 |
+
"model.layers.28.self_attn.o_proj",
|
| 289 |
+
"model.layers.28.self_attn.gate_proj",
|
| 290 |
+
"model.layers.28.mlp.router.gate",
|
| 291 |
+
"model.layers.29.self_attn.q_proj",
|
| 292 |
+
"model.layers.29.self_attn.k_proj",
|
| 293 |
+
"model.layers.29.self_attn.v_proj",
|
| 294 |
+
"model.layers.29.self_attn.o_proj",
|
| 295 |
+
"model.layers.29.self_attn.gate_proj",
|
| 296 |
+
"model.layers.29.mlp.router.gate",
|
| 297 |
+
"model.layers.30.self_attn.q_proj",
|
| 298 |
+
"model.layers.30.self_attn.k_proj",
|
| 299 |
+
"model.layers.30.self_attn.v_proj",
|
| 300 |
+
"model.layers.30.self_attn.o_proj",
|
| 301 |
+
"model.layers.30.self_attn.gate_proj",
|
| 302 |
+
"model.layers.30.mlp.router.gate",
|
| 303 |
+
"model.layers.31.self_attn.q_proj",
|
| 304 |
+
"model.layers.31.self_attn.k_proj",
|
| 305 |
+
"model.layers.31.self_attn.v_proj",
|
| 306 |
+
"model.layers.31.self_attn.o_proj",
|
| 307 |
+
"model.layers.31.self_attn.gate_proj",
|
| 308 |
+
"model.layers.31.mlp.router.gate",
|
| 309 |
+
"model.layers.32.self_attn.q_proj",
|
| 310 |
+
"model.layers.32.self_attn.k_proj",
|
| 311 |
+
"model.layers.32.self_attn.v_proj",
|
| 312 |
+
"model.layers.32.self_attn.o_proj",
|
| 313 |
+
"model.layers.32.self_attn.gate_proj",
|
| 314 |
+
"model.layers.32.mlp.router.gate",
|
| 315 |
+
"model.layers.33.self_attn.q_proj",
|
| 316 |
+
"model.layers.33.self_attn.k_proj",
|
| 317 |
+
"model.layers.33.self_attn.v_proj",
|
| 318 |
+
"model.layers.33.self_attn.o_proj",
|
| 319 |
+
"model.layers.33.self_attn.gate_proj",
|
| 320 |
+
"model.layers.33.mlp.router.gate",
|
| 321 |
+
"model.layers.34.self_attn.q_proj",
|
| 322 |
+
"model.layers.34.self_attn.k_proj",
|
| 323 |
+
"model.layers.34.self_attn.v_proj",
|
| 324 |
+
"model.layers.34.self_attn.o_proj",
|
| 325 |
+
"model.layers.34.self_attn.gate_proj",
|
| 326 |
+
"model.layers.34.mlp.router.gate",
|
| 327 |
+
"model.layers.35.self_attn.q_proj",
|
| 328 |
+
"model.layers.35.self_attn.k_proj",
|
| 329 |
+
"model.layers.35.self_attn.v_proj",
|
| 330 |
+
"model.layers.35.self_attn.o_proj",
|
| 331 |
+
"model.layers.35.self_attn.gate_proj",
|
| 332 |
+
"model.layers.35.mlp.router.gate",
|
| 333 |
+
"model.layers.36.self_attn.q_proj",
|
| 334 |
+
"model.layers.36.self_attn.k_proj",
|
| 335 |
+
"model.layers.36.self_attn.v_proj",
|
| 336 |
+
"model.layers.36.self_attn.o_proj",
|
| 337 |
+
"model.layers.36.self_attn.gate_proj",
|
| 338 |
+
"model.layers.36.mlp.router.gate",
|
| 339 |
+
"model.layers.37.self_attn.q_proj",
|
| 340 |
+
"model.layers.37.self_attn.k_proj",
|
| 341 |
+
"model.layers.37.self_attn.v_proj",
|
| 342 |
+
"model.layers.37.self_attn.o_proj",
|
| 343 |
+
"model.layers.37.self_attn.gate_proj",
|
| 344 |
+
"model.layers.37.mlp.router.gate",
|
| 345 |
+
"model.layers.38.self_attn.q_proj",
|
| 346 |
+
"model.layers.38.self_attn.k_proj",
|
| 347 |
+
"model.layers.38.self_attn.v_proj",
|
| 348 |
+
"model.layers.38.self_attn.o_proj",
|
| 349 |
+
"model.layers.38.self_attn.gate_proj",
|
| 350 |
+
"model.layers.38.mlp.router.gate",
|
| 351 |
+
"model.layers.39.self_attn.q_proj",
|
| 352 |
+
"model.layers.39.self_attn.k_proj",
|
| 353 |
+
"model.layers.39.self_attn.v_proj",
|
| 354 |
+
"model.layers.39.self_attn.o_proj",
|
| 355 |
+
"model.layers.39.self_attn.gate_proj",
|
| 356 |
+
"model.layers.39.mlp.router.gate",
|
| 357 |
+
"model.layers.40.self_attn.q_proj",
|
| 358 |
+
"model.layers.40.self_attn.k_proj",
|
| 359 |
+
"model.layers.40.self_attn.v_proj",
|
| 360 |
+
"model.layers.40.self_attn.o_proj",
|
| 361 |
+
"model.layers.40.self_attn.gate_proj",
|
| 362 |
+
"model.layers.40.mlp.router.gate",
|
| 363 |
+
"model.layers.41.self_attn.q_proj",
|
| 364 |
+
"model.layers.41.self_attn.k_proj",
|
| 365 |
+
"model.layers.41.self_attn.v_proj",
|
| 366 |
+
"model.layers.41.self_attn.o_proj",
|
| 367 |
+
"model.layers.41.self_attn.gate_proj",
|
| 368 |
+
"model.layers.41.mlp.router.gate",
|
| 369 |
+
"model.layers.42.self_attn.q_proj",
|
| 370 |
+
"model.layers.42.self_attn.k_proj",
|
| 371 |
+
"model.layers.42.self_attn.v_proj",
|
| 372 |
+
"model.layers.42.self_attn.o_proj",
|
| 373 |
+
"model.layers.42.self_attn.gate_proj",
|
| 374 |
+
"model.layers.42.mlp.router.gate",
|
| 375 |
+
"model.layers.43.self_attn.q_proj",
|
| 376 |
+
"model.layers.43.self_attn.k_proj",
|
| 377 |
+
"model.layers.43.self_attn.v_proj",
|
| 378 |
+
"model.layers.43.self_attn.o_proj",
|
| 379 |
+
"model.layers.43.self_attn.gate_proj",
|
| 380 |
+
"model.layers.43.mlp.router.gate",
|
| 381 |
+
"model.layers.44.self_attn.q_proj",
|
| 382 |
+
"model.layers.44.self_attn.k_proj",
|
| 383 |
+
"model.layers.44.self_attn.v_proj",
|
| 384 |
+
"model.layers.44.self_attn.o_proj",
|
| 385 |
+
"model.layers.44.self_attn.gate_proj",
|
| 386 |
+
"model.layers.44.mlp.router.gate",
|
| 387 |
+
"model.layers.45.self_attn.q_proj",
|
| 388 |
+
"model.layers.45.self_attn.k_proj",
|
| 389 |
+
"model.layers.45.self_attn.v_proj",
|
| 390 |
+
"model.layers.45.self_attn.o_proj",
|
| 391 |
+
"model.layers.45.self_attn.gate_proj",
|
| 392 |
+
"model.layers.45.mlp.router.gate",
|
| 393 |
+
"model.layers.46.self_attn.q_proj",
|
| 394 |
+
"model.layers.46.self_attn.k_proj",
|
| 395 |
+
"model.layers.46.self_attn.v_proj",
|
| 396 |
+
"model.layers.46.self_attn.o_proj",
|
| 397 |
+
"model.layers.46.self_attn.gate_proj",
|
| 398 |
+
"model.layers.46.mlp.router.gate",
|
| 399 |
+
"model.layers.47.self_attn.q_proj",
|
| 400 |
+
"model.layers.47.self_attn.k_proj",
|
| 401 |
+
"model.layers.47.self_attn.v_proj",
|
| 402 |
+
"model.layers.47.self_attn.o_proj",
|
| 403 |
+
"model.layers.47.self_attn.gate_proj",
|
| 404 |
+
"model.layers.47.mlp.router.gate",
|
| 405 |
+
"model.layers.48.self_attn.q_proj",
|
| 406 |
+
"model.layers.48.self_attn.k_proj",
|
| 407 |
+
"model.layers.48.self_attn.v_proj",
|
| 408 |
+
"model.layers.48.self_attn.o_proj",
|
| 409 |
+
"model.layers.48.self_attn.gate_proj",
|
| 410 |
+
"model.layers.48.mlp.router.gate",
|
| 411 |
+
"model.layers.49.self_attn.q_proj",
|
| 412 |
+
"model.layers.49.self_attn.k_proj",
|
| 413 |
+
"model.layers.49.self_attn.v_proj",
|
| 414 |
+
"model.layers.49.self_attn.o_proj",
|
| 415 |
+
"model.layers.49.self_attn.gate_proj",
|
| 416 |
+
"model.layers.49.mlp.router.gate",
|
| 417 |
+
"model.layers.50.self_attn.q_proj",
|
| 418 |
+
"model.layers.50.self_attn.k_proj",
|
| 419 |
+
"model.layers.50.self_attn.v_proj",
|
| 420 |
+
"model.layers.50.self_attn.o_proj",
|
| 421 |
+
"model.layers.50.self_attn.gate_proj",
|
| 422 |
+
"model.layers.50.mlp.router.gate",
|
| 423 |
+
"model.layers.51.self_attn.q_proj",
|
| 424 |
+
"model.layers.51.self_attn.k_proj",
|
| 425 |
+
"model.layers.51.self_attn.v_proj",
|
| 426 |
+
"model.layers.51.self_attn.o_proj",
|
| 427 |
+
"model.layers.51.self_attn.gate_proj",
|
| 428 |
+
"model.layers.51.mlp.router.gate",
|
| 429 |
+
"model.layers.52.self_attn.q_proj",
|
| 430 |
+
"model.layers.52.self_attn.k_proj",
|
| 431 |
+
"model.layers.52.self_attn.v_proj",
|
| 432 |
+
"model.layers.52.self_attn.o_proj",
|
| 433 |
+
"model.layers.52.self_attn.gate_proj",
|
| 434 |
+
"model.layers.52.mlp.router.gate",
|
| 435 |
+
"model.layers.53.self_attn.q_proj",
|
| 436 |
+
"model.layers.53.self_attn.k_proj",
|
| 437 |
+
"model.layers.53.self_attn.v_proj",
|
| 438 |
+
"model.layers.53.self_attn.o_proj",
|
| 439 |
+
"model.layers.53.self_attn.gate_proj",
|
| 440 |
+
"model.layers.53.mlp.router.gate",
|
| 441 |
+
"model.layers.54.self_attn.q_proj",
|
| 442 |
+
"model.layers.54.self_attn.k_proj",
|
| 443 |
+
"model.layers.54.self_attn.v_proj",
|
| 444 |
+
"model.layers.54.self_attn.o_proj",
|
| 445 |
+
"model.layers.54.self_attn.gate_proj",
|
| 446 |
+
"model.layers.54.mlp.router.gate",
|
| 447 |
+
"model.layers.55.self_attn.q_proj",
|
| 448 |
+
"model.layers.55.self_attn.k_proj",
|
| 449 |
+
"model.layers.55.self_attn.v_proj",
|
| 450 |
+
"model.layers.55.self_attn.o_proj",
|
| 451 |
+
"model.layers.55.self_attn.gate_proj",
|
| 452 |
+
"model.layers.55.mlp.router.gate",
|
| 453 |
+
"model.layers.56.self_attn.q_proj",
|
| 454 |
+
"model.layers.56.self_attn.k_proj",
|
| 455 |
+
"model.layers.56.self_attn.v_proj",
|
| 456 |
+
"model.layers.56.self_attn.o_proj",
|
| 457 |
+
"model.layers.56.self_attn.gate_proj",
|
| 458 |
+
"model.layers.56.mlp.router.gate",
|
| 459 |
+
"model.layers.57.self_attn.q_proj",
|
| 460 |
+
"model.layers.57.self_attn.k_proj",
|
| 461 |
+
"model.layers.57.self_attn.v_proj",
|
| 462 |
+
"model.layers.57.self_attn.o_proj",
|
| 463 |
+
"model.layers.57.self_attn.gate_proj",
|
| 464 |
+
"model.layers.57.mlp.router.gate",
|
| 465 |
+
"model.layers.58.self_attn.q_proj",
|
| 466 |
+
"model.layers.58.self_attn.k_proj",
|
| 467 |
+
"model.layers.58.self_attn.v_proj",
|
| 468 |
+
"model.layers.58.self_attn.o_proj",
|
| 469 |
+
"model.layers.58.self_attn.gate_proj",
|
| 470 |
+
"model.layers.58.mlp.router.gate",
|
| 471 |
+
"model.layers.59.self_attn.q_proj",
|
| 472 |
+
"model.layers.59.self_attn.k_proj",
|
| 473 |
+
"model.layers.59.self_attn.v_proj",
|
| 474 |
+
"model.layers.59.self_attn.o_proj",
|
| 475 |
+
"model.layers.59.self_attn.gate_proj",
|
| 476 |
+
"model.layers.59.mlp.router.gate",
|
| 477 |
+
"lm_head"
|
| 478 |
+
],
|
| 479 |
+
"kv_cache_scheme": null,
|
| 480 |
+
"quant_method": "compressed-tensors",
|
| 481 |
+
"quantization_status": "compressed",
|
| 482 |
+
"sparsity_config": {},
|
| 483 |
+
"transform_config": {},
|
| 484 |
+
"version": "0.14.0"
|
| 485 |
+
},
|
| 486 |
+
"rms_norm_eps": 1e-05,
|
| 487 |
+
"rope_scaling": null,
|
| 488 |
+
"rope_theta": 10000,
|
| 489 |
+
"route_norm": true,
|
| 490 |
+
"route_scale": 2.448,
|
| 491 |
+
"score_func": "sigmoid",
|
| 492 |
+
"sliding_window": 4096,
|
| 493 |
+
"tie_word_embeddings": false,
|
| 494 |
+
"topk_group": 1,
|
| 495 |
+
"transformers_version": "4.57.6",
|
| 496 |
+
"use_cache": true,
|
| 497 |
+
"use_grouped_mm": true,
|
| 498 |
+
"vocab_size": 200192
|
| 499 |
+
}
|
configuration_afmoe.py
ADDED
|
@@ -0,0 +1,133 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# coding=utf-8
|
| 2 |
+
# Copyright 2022 EleutherAI and the HuggingFace Inc. team. All rights reserved.
|
| 3 |
+
#
|
| 4 |
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
| 5 |
+
# you may not use this file except in compliance with the License.
|
| 6 |
+
# You may obtain a copy of the License at
|
| 7 |
+
#
|
| 8 |
+
# http://www.apache.org/licenses/LICENSE-2.0
|
| 9 |
+
#
|
| 10 |
+
# Unless required by applicable law or agreed to in writing, software
|
| 11 |
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
| 12 |
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| 13 |
+
# See the License for the specific language governing permissions and
|
| 14 |
+
# limitations under the License.
|
| 15 |
+
from transformers.configuration_utils import PretrainedConfig
|
| 16 |
+
from transformers.modeling_rope_utils import rope_config_validation
|
| 17 |
+
from transformers.configuration_utils import layer_type_validation
|
| 18 |
+
from transformers.utils import logging
|
| 19 |
+
|
| 20 |
+
logger = logging.get_logger(__name__)
|
| 21 |
+
|
| 22 |
+
class AfmoeConfig(PretrainedConfig):
|
| 23 |
+
"""
|
| 24 |
+
n_group (`int`, *optional*, defaults to 1):
|
| 25 |
+
Number of groups for routed experts.
|
| 26 |
+
topk_group (`int`, *optional*, defaults to 1):
|
| 27 |
+
Number of selected groups for each token(for each token, ensuring the selected experts is only within `topk_group` groups).
|
| 28 |
+
"""
|
| 29 |
+
model_type = "afmoe"
|
| 30 |
+
base_model_pp_plan = {
|
| 31 |
+
"embed_tokens": (["input_ids"], ["inputs_embeds"]),
|
| 32 |
+
"layers": (["hidden_states", "attention_mask"], ["hidden_states"]),
|
| 33 |
+
"norm": (["hidden_states"], ["hidden_states"]),
|
| 34 |
+
}
|
| 35 |
+
|
| 36 |
+
def __init__(
|
| 37 |
+
self,
|
| 38 |
+
num_hidden_layers: int = 32,
|
| 39 |
+
vocab_size: int = 200192,
|
| 40 |
+
hidden_size: int = 2048,
|
| 41 |
+
intermediate_size: int = 6144,
|
| 42 |
+
moe_intermediate_size=1408,
|
| 43 |
+
num_dense_layers=1,
|
| 44 |
+
num_attention_heads=16,
|
| 45 |
+
num_key_value_heads=None,
|
| 46 |
+
head_dim=128,
|
| 47 |
+
hidden_act="silu",
|
| 48 |
+
max_position_embeddings=16384,
|
| 49 |
+
initializer_range=0.02,
|
| 50 |
+
rms_norm_eps=1e-5,
|
| 51 |
+
use_cache=True,
|
| 52 |
+
tie_word_embeddings=False,
|
| 53 |
+
rope_theta=10000.0,
|
| 54 |
+
rope_scaling=None,
|
| 55 |
+
num_experts=64,
|
| 56 |
+
num_experts_per_tok=6,
|
| 57 |
+
num_shared_experts=2,
|
| 58 |
+
num_expert_groups=1,
|
| 59 |
+
num_limited_groups=1,
|
| 60 |
+
score_func="sigmoid",
|
| 61 |
+
route_norm=True,
|
| 62 |
+
route_scale=1.0,
|
| 63 |
+
global_attn_every_n_layers=4,
|
| 64 |
+
sliding_window=1024,
|
| 65 |
+
mup_enabled=False,
|
| 66 |
+
layer_types=None,
|
| 67 |
+
attention_dropout: float = 0.0,
|
| 68 |
+
n_group: int = 1,
|
| 69 |
+
topk_group: int = 1,
|
| 70 |
+
**kwargs,
|
| 71 |
+
):
|
| 72 |
+
self.vocab_size = vocab_size
|
| 73 |
+
self.max_position_embeddings = max_position_embeddings
|
| 74 |
+
self.hidden_size = hidden_size
|
| 75 |
+
self.intermediate_size = intermediate_size
|
| 76 |
+
self.num_hidden_layers = num_hidden_layers
|
| 77 |
+
self.num_dense_layers = num_dense_layers
|
| 78 |
+
self.num_attention_heads = num_attention_heads
|
| 79 |
+
self.head_dim = head_dim
|
| 80 |
+
self.hidden_act = hidden_act
|
| 81 |
+
self.initializer_range = initializer_range
|
| 82 |
+
self.rms_norm_eps = rms_norm_eps
|
| 83 |
+
self.use_cache = use_cache
|
| 84 |
+
self.rope_theta = rope_theta
|
| 85 |
+
self.rope_scaling = rope_scaling
|
| 86 |
+
|
| 87 |
+
|
| 88 |
+
# MoE specific
|
| 89 |
+
self.moe_intermediate_size = moe_intermediate_size
|
| 90 |
+
self.num_experts_per_tok = num_experts_per_tok
|
| 91 |
+
self.n_group = n_group
|
| 92 |
+
self.topk_group = topk_group
|
| 93 |
+
self.num_experts = num_experts
|
| 94 |
+
self.num_shared_experts = num_shared_experts
|
| 95 |
+
self.num_expert_groups = num_expert_groups
|
| 96 |
+
self.num_limited_groups = num_limited_groups
|
| 97 |
+
self.score_func = score_func
|
| 98 |
+
self.route_norm = route_norm
|
| 99 |
+
self.route_scale = route_scale
|
| 100 |
+
|
| 101 |
+
|
| 102 |
+
# Attention specific
|
| 103 |
+
self.attention_dropout = attention_dropout
|
| 104 |
+
self.global_attn_every_n_layers = global_attn_every_n_layers
|
| 105 |
+
self.sliding_window = sliding_window
|
| 106 |
+
self.layer_types = layer_types
|
| 107 |
+
if self.layer_types is None:
|
| 108 |
+
self.layer_types = [
|
| 109 |
+
"sliding_attention" if bool((i + 1) % global_attn_every_n_layers) else "full_attention" for i in range(self.num_hidden_layers)
|
| 110 |
+
]
|
| 111 |
+
layer_type_validation(self.layer_types)
|
| 112 |
+
|
| 113 |
+
# muP specific
|
| 114 |
+
self.mup_enabled = mup_enabled
|
| 115 |
+
|
| 116 |
+
if num_key_value_heads is None:
|
| 117 |
+
num_key_value_heads = num_attention_heads
|
| 118 |
+
|
| 119 |
+
self.num_key_value_heads = num_key_value_heads
|
| 120 |
+
|
| 121 |
+
|
| 122 |
+
# Validate rope configs
|
| 123 |
+
if self.rope_scaling is not None and "type" in self.rope_scaling:
|
| 124 |
+
self.rope_scaling["rope_type"] = self.rope_scaling["type"]
|
| 125 |
+
rope_config_validation(self)
|
| 126 |
+
|
| 127 |
+
super().__init__(
|
| 128 |
+
tie_word_embeddings=tie_word_embeddings,
|
| 129 |
+
**kwargs,
|
| 130 |
+
)
|
| 131 |
+
|
| 132 |
+
|
| 133 |
+
__all__ = ["AfmoeConfig"]
|
generation_config.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_from_model_config": true,
|
| 3 |
+
"bos_token_id": 0,
|
| 4 |
+
"eos_token_id": 3,
|
| 5 |
+
"pad_token_id": 12,
|
| 6 |
+
"transformers_version": "4.57.6"
|
| 7 |
+
}
|
model-00001-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c82a00bef1546064c3b88f8be8e6d4d5f23376d8801f4f82a380c23758250167
|
| 3 |
+
size 4998314816
|
model-00002-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4e11487095d803e79a894ccf57e58d76e4477b10b9afbba45493cf448eb11e56
|
| 3 |
+
size 4998729232
|
model-00003-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ddb52ab10b4eaed26483e6944c820639385773a02afbccfb3294d27f8dec4be8
|
| 3 |
+
size 4999630856
|
model-00004-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06326ee70516878445fb155ba363ece0159af5a8d092431916b0663cc19232e2
|
| 3 |
+
size 4998730128
|
model-00005-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f81ac24a55b59fd4e68f72efa136ab6f394ae72cfbe1fd2b64db9ba2b2562c99
|
| 3 |
+
size 4998732240
|
model-00006-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:060f40ab2b34dd94bc1b468bff1d8023fa3d126fba206b10d72c0eda2fcee361
|
| 3 |
+
size 4999634128
|
model-00007-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c2b48f364905c9985831ceee8a45ff996980932fb6db0b0d560ca588e2b51215
|
| 3 |
+
size 4998731568
|
model-00008-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c2c1bffadaa5ddb691f99b141143b05cbf4deebc46c4de07e3de727e695d2a46
|
| 3 |
+
size 4998732240
|
model-00009-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d3ca009946777571a51c6c76fdf08f5c45179b43fc91dd09b5d65d60819aac40
|
| 3 |
+
size 4998732240
|
model-00010-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bc53ab813351d8fbf31cf78f1eaeba3370d40a120b9f3b5b895b565e6210ef2d
|
| 3 |
+
size 4999633760
|
model-00011-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4644f290a9942ad3081b5f14a1a2921aba55a96be8bf3a6b343064cf2b55e095
|
| 3 |
+
size 4998731936
|
model-00012-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:26fc7185b7ede8ed96ecdf4e32e528982b2badf5bdd189c8cf803a33c4345569
|
| 3 |
+
size 4998732240
|
model-00013-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4b2c9c5454a53525e6e2a4b41c41ce2e979b9585c592761874ec0003f105bdc3
|
| 3 |
+
size 4999634080
|
model-00014-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dbda7bb571d5698ca4dcda9f08b302abd9fa956c1315b9732cedc3bf0c50615e
|
| 3 |
+
size 4998731608
|
model-00015-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:223490e83202a68dc45d72c031c002308ee7571520cd4582ec2f5f0e3a4cb59c
|
| 3 |
+
size 4998732248
|
model-00016-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f3ad2053bd4396f60f6d75c4ff43e4e0bf8c7ae8ba51d9da7bc6c661630833c5
|
| 3 |
+
size 4990666848
|
model-00017-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f1f216f4fb713a0ec6bf82b70de96fc21269189bb8d432c98fbc109feda5a407
|
| 3 |
+
size 4997966232
|
model-00018-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:752362781866e595597bbdc47418820729fb20204ef0328c198d513d7cb07322
|
| 3 |
+
size 4998731976
|
model-00019-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8c683ccf4474855193a6ce7e32971810a11a38e9eef73a6b39c43b10c3a5e560
|
| 3 |
+
size 4998732240
|
model-00020-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1c662d2c89e96e9887e746e86fd86c1c011c6cbd6714cfc6d164f033dde35ebd
|
| 3 |
+
size 4999634040
|
model-00021-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8fd1d5cbb429a4f5c11efa50c2a3ca0e2fb70ff48307c52d6a931c4774496eed
|
| 3 |
+
size 4998731656
|
model-00022-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f8df71ef2da92c8deacd006f4bd9afb433d33124ff6f0f7ba0e49151f3a3fe2d
|
| 3 |
+
size 4998732240
|
model-00023-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0b0cfee53437735a650a1fa286f0904b76ca9ad07781aa1afb8f94508c335a5b
|
| 3 |
+
size 4999634416
|
model-00024-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fb5a25b90c71d3d23a0e8c66ec9f9a23ca923b16e6ae28ecc9e77fc85732aa30
|
| 3 |
+
size 4998731496
|
model-00025-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b6e1350de169cb53f4e1ffc4fe8e5a324dc5094d92a8a8efdb8575849f587db9
|
| 3 |
+
size 4998732024
|
model-00026-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ae54f7987c2cfd5117e79b518a8a867cce83046174b78df104cc83a4e34ab6e
|
| 3 |
+
size 4998732240
|
model-00027-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:76526ef2fc11cebc28a5ffd8dbcea920c721991f5dadd2547d31a239de06cd12
|
| 3 |
+
size 4999633992
|
model-00028-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ed3a33f26f006cf6e105591537b0c509040dc364019698e8e49a2d4c886bc4c1
|
| 3 |
+
size 4998731704
|
model-00029-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2d5f91aa0172033758bed144e6a1fdf9db163bd426a4ea6abb9a5fb8cf8d324a
|
| 3 |
+
size 4998732248
|
model-00030-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4094963b2588eea596816c4f1ab6ae4e5b37268ad152dfb6b4ddf2573bc286e2
|
| 3 |
+
size 4999634360
|
model-00031-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3b535adf1c08a9100501c98e33697d19ab8b6d19e52986bc49d4e4deb6f40f77
|
| 3 |
+
size 4998731504
|
model-00032-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c846227070eeb45808fe20cd0c8c2f1dec96a59c3325e21d9713173020fd8630
|
| 3 |
+
size 4998732072
|
model-00033-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:372195e108773fb3485340540fe52658495a49db93c6df7bb2b463321d71dda6
|
| 3 |
+
size 4998732248
|
model-00034-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8f0315c5be778de048f8c9f2293d3c25898c6e42060ad933a5a6c7e4d2749d57
|
| 3 |
+
size 4999633944
|
model-00035-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4b66a990468df7f5f7ada7c1d8457fc66fdfb434002360687c26c96843592c32
|
| 3 |
+
size 4998731752
|
model-00036-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f8f625f5277620659050ecbbb1e2514036922fd173f7ce0794d684b9adf4372
|
| 3 |
+
size 4998732240
|
model-00037-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:97b589d5b824c87bd39f9a05e0500c62c05b6f687fe4eba8832d9ba73669b08c
|
| 3 |
+
size 4999634264
|
model-00038-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3db5c4f34e5cb6897513f4b0d5db3ce6161e4a53d4335076aed2b6b0bbffee4a
|
| 3 |
+
size 4998731552
|
model-00039-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4389cb99bc63fc39557b3432bc42ae7ceb84bb43a75525027cef280f722d91e6
|
| 3 |
+
size 4998732120
|
model-00040-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:382653a93ea7a62dc4fd98236c98a1f114d8cf6d4cc30bad81da7e70acd242de
|
| 3 |
+
size 4998732240
|
model-00041-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:16b8ab79b338915e836462f59ce5289ef0a99b3c14863bd27992265ac5b2368c
|
| 3 |
+
size 4999633896
|
model-00042-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f261fb5412bbfb70f822da06aa49ec977c2dfb93da3a5744884e10d6e528f66e
|
| 3 |
+
size 4998731800
|
model-00043-of-00043.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:32fadb52c42721744d1e30068c8cd86ae39c6a3b69641b498b5c299e3b23f106
|
| 3 |
+
size 3108429960
|