Instructions to use mlx-community/SongGeneration-v2-medium-bf16 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use mlx-community/SongGeneration-v2-medium-bf16 with MLX:
# Download the model from the Hub pip install huggingface_hub[hf_xet] huggingface-cli download --local-dir SongGeneration-v2-medium-bf16 mlx-community/SongGeneration-v2-medium-bf16
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- LM Studio
Add files using upload-large-folder tool
Browse files- README.md +114 -0
- config.json +129 -0
- config.official.yaml +141 -0
- mlx_manifest.json +129 -0
- model-00002-of-00094.safetensors +3 -0
- model-00003-of-00094.safetensors +3 -0
- model-00006-of-00094.safetensors +3 -0
- model-00007-of-00094.safetensors +3 -0
- model-00008-of-00094.safetensors +3 -0
- model-00009-of-00094.safetensors +3 -0
- model-00011-of-00094.safetensors +3 -0
- model-00014-of-00094.safetensors +3 -0
- model-00015-of-00094.safetensors +3 -0
- model-00022-of-00094.safetensors +3 -0
- model-00023-of-00094.safetensors +3 -0
- model-00026-of-00094.safetensors +3 -0
- model-00027-of-00094.safetensors +3 -0
- model-00030-of-00094.safetensors +3 -0
- model-00031-of-00094.safetensors +3 -0
- model-00034-of-00094.safetensors +3 -0
- model-00035-of-00094.safetensors +3 -0
- model-00042-of-00094.safetensors +3 -0
- model-00043-of-00094.safetensors +3 -0
- model-00046-of-00094.safetensors +3 -0
- model-00047-of-00094.safetensors +3 -0
- model-00048-of-00094.safetensors +3 -0
- model-00049-of-00094.safetensors +3 -0
- model-00050-of-00094.safetensors +3 -0
- model-00054-of-00094.safetensors +3 -0
- model-00055-of-00094.safetensors +3 -0
- model-00062-of-00094.safetensors +3 -0
- model-00063-of-00094.safetensors +3 -0
- model-00066-of-00094.safetensors +3 -0
- model-00067-of-00094.safetensors +3 -0
- model-00068-of-00094.safetensors +3 -0
- model-00070-of-00094.safetensors +3 -0
- model-00071-of-00094.safetensors +3 -0
- model-00074-of-00094.safetensors +3 -0
- model-00075-of-00094.safetensors +3 -0
- model-00082-of-00094.safetensors +3 -0
- model-00083-of-00094.safetensors +3 -0
- model-00086-of-00094.safetensors +3 -0
- model-00087-of-00094.safetensors +3 -0
- model-00088-of-00094.safetensors +3 -0
- model-00089-of-00094.safetensors +3 -0
- model-00090-of-00094.safetensors +3 -0
- model-00091-of-00094.safetensors +3 -0
- model-00094-of-00094.safetensors +3 -0
- model.safetensors.index.json +390 -0
- vocab.yaml +13 -0
README.md
ADDED
|
@@ -0,0 +1,114 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: other
|
| 3 |
+
library_name: mlx
|
| 4 |
+
pipeline_tag: text-to-audio
|
| 5 |
+
base_model:
|
| 6 |
+
- tencent/SongGeneration
|
| 7 |
+
tags:
|
| 8 |
+
- mlx
|
| 9 |
+
- apple-silicon
|
| 10 |
+
- music-generation
|
| 11 |
+
- song-generation
|
| 12 |
+
- audiolm
|
| 13 |
+
---
|
| 14 |
+
|
| 15 |
+
Part of the SongGeneration MLX conversion set.
|
| 16 |
+
|
| 17 |
+
# SongGeneration-v2-medium-bfloat16
|
| 18 |
+
|
| 19 |
+
Apple MLX weights for the autoregressive `audiolm` token generator from Tencent SongGeneration v2-medium.
|
| 20 |
+
|
| 21 |
+
This is not a full-stack pure MLX audio pipeline yet: token generation runs with MLX, while FLAC decoding currently uses the official PyTorch Flow1dVAE / separate-tokenizer bridge in [`ailuntx/SongGeneration-MLX`](https://github.com/ailuntx/SongGeneration-MLX).
|
| 22 |
+
|
| 23 |
+
## TL;DR
|
| 24 |
+
|
| 25 |
+
| | |
|
| 26 |
+
|---|---|
|
| 27 |
+
| **Variant** | `v2-medium` |
|
| 28 |
+
| **Precision** | `bfloat16` |
|
| 29 |
+
| **Converted component** | SongGeneration `audiolm` token generator |
|
| 30 |
+
| **Runtime** | [`ailuntx/SongGeneration-MLX`](https://github.com/ailuntx/SongGeneration-MLX) |
|
| 31 |
+
| **Official model** | [`tencent/SongGeneration`](https://huggingface.co/tencent/SongGeneration) |
|
| 32 |
+
| **Official code** | [`tencent-ailab/songgeneration`](https://github.com/tencent-ailab/songgeneration) |
|
| 33 |
+
|
| 34 |
+
## Quick Start
|
| 35 |
+
|
| 36 |
+
```bash
|
| 37 |
+
hf download mlx-community/SongGeneration-v2-medium-bfloat16 --local-dir ./models/SongGeneration-v2-medium-bfloat16
|
| 38 |
+
|
| 39 |
+
git clone https://github.com/ailuntx/SongGeneration-MLX.git
|
| 40 |
+
cd SongGeneration-MLX
|
| 41 |
+
python -m venv .venv
|
| 42 |
+
.venv/bin/pip install -e .
|
| 43 |
+
|
| 44 |
+
PYTHONPATH=. .venv/bin/python -m songgeneration_mlx.cli \
|
| 45 |
+
--model ./models/SongGeneration-v2-medium-bfloat16 \
|
| 46 |
+
--lyrics '[verse] hello from mlx [chorus] sing it again' \
|
| 47 |
+
--description 'Pop, female vocal, bright production, [Musicality-medium].' \
|
| 48 |
+
--duration 2 \
|
| 49 |
+
--top-k 50 \
|
| 50 |
+
--temperature 0.9 \
|
| 51 |
+
--output tokens_2s.npz
|
| 52 |
+
```
|
| 53 |
+
|
| 54 |
+
To decode tokens to FLAC, use the official decoder bridge documented in the GitHub README. The bridge requires the official SongGeneration runtime assets.
|
| 55 |
+
|
| 56 |
+
## Variants
|
| 57 |
+
|
| 58 |
+
| Variant | Disk | Notes |
|
| 59 |
+
|---|---:|---|
|
| 60 |
+
| `SongGeneration-v2-medium-fp32` | 10G | high-precision medium baseline |
|
| 61 |
+
| `SongGeneration-v2-medium-bfloat16` | 5.2G | recommended medium quality baseline |
|
| 62 |
+
| `SongGeneration-v2-medium-8bit` | 2.8G | smaller medium checkpoint |
|
| 63 |
+
| `SongGeneration-v2-medium-4bit` | 1.5G | smallest medium checkpoint |
|
| 64 |
+
| `SongGeneration-v2-large-fp32` | 19G | high-precision large baseline |
|
| 65 |
+
| `SongGeneration-v2-large-bfloat16` | 9.5G | large quality baseline |
|
| 66 |
+
| `SongGeneration-v2-large-8bit` | 5.0G | smaller large checkpoint |
|
| 67 |
+
| `SongGeneration-v2-large-4bit` | 2.7G | smallest large checkpoint |
|
| 68 |
+
|
| 69 |
+
## Layout
|
| 70 |
+
|
| 71 |
+
```text
|
| 72 |
+
SongGeneration-v2-medium-bfloat16/
|
| 73 |
+
|-- model-00001-of-000xx.safetensors
|
| 74 |
+
|-- model.safetensors.index.json
|
| 75 |
+
|-- config.json
|
| 76 |
+
|-- mlx_manifest.json
|
| 77 |
+
|-- config.official.yaml
|
| 78 |
+
|-- vocab.yaml
|
| 79 |
+
`-- qwen2_tokenizer/
|
| 80 |
+
```
|
| 81 |
+
|
| 82 |
+
## Validation
|
| 83 |
+
|
| 84 |
+
Local Apple Silicon validation was run on the medium bfloat16 path:
|
| 85 |
+
|
| 86 |
+
| Test | Result |
|
| 87 |
+
|---|---|
|
| 88 |
+
| 12s MLX token generation | 550 pattern steps, about 1 minute wall time |
|
| 89 |
+
| 12s official decoder bridge | 73.27s wall time |
|
| 90 |
+
| 12s FLAC | 48kHz stereo, 12.000s, RMS about `0.163` |
|
| 91 |
+
|
| 92 |
+
The official recent-token repetition penalty is implemented in the MLX runtime. Without it, 12s generations collapse into repeated tokens and decode close to silence.
|
| 93 |
+
|
| 94 |
+
## License
|
| 95 |
+
|
| 96 |
+
License follows the upstream SongGeneration release. Check the official model card and repository for the authoritative model license.
|
| 97 |
+
|
| 98 |
+
## Citation
|
| 99 |
+
|
| 100 |
+
```bibtex
|
| 101 |
+
@misc{songgeneration-mlx,
|
| 102 |
+
title = {SongGeneration-MLX: Apple MLX port of SongGeneration},
|
| 103 |
+
author = {ailuntx},
|
| 104 |
+
year = {2026},
|
| 105 |
+
url = {https://github.com/ailuntx/SongGeneration-MLX},
|
| 106 |
+
}
|
| 107 |
+
|
| 108 |
+
@article{lei2025levo,
|
| 109 |
+
title = {LeVo: High-Quality Song Generation with Multi-Preference Alignment},
|
| 110 |
+
author = {Lei, Shun and Xu, Yaoxun and Lin, Zhiwei and Zhang, Huaicheng and Tan, Wei and Chen, Hangting and Yu, Jianwei and Zhang, Yixuan and Yang, Chenyu and Zhu, Haina and Wang, Shuai and Wu, Zhiyong and Yu, Dong},
|
| 111 |
+
journal = {arXiv preprint arXiv:2506.07520},
|
| 112 |
+
year = {2025},
|
| 113 |
+
}
|
| 114 |
+
```
|
config.json
ADDED
|
@@ -0,0 +1,129 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"format": "songgeneration-mlx-audiolm",
|
| 3 |
+
"source": "tencent/SongGeneration",
|
| 4 |
+
"official_code": "https://github.com/tencent-ailab/songgeneration",
|
| 5 |
+
"variant": "v2-medium",
|
| 6 |
+
"precision": "bfloat16",
|
| 7 |
+
"runtime": {
|
| 8 |
+
"hidden_size": 1536,
|
| 9 |
+
"intermediate_size": 8960,
|
| 10 |
+
"num_heads": 12,
|
| 11 |
+
"num_layers": 28,
|
| 12 |
+
"num_layers_sub": 12,
|
| 13 |
+
"code_depth": 3,
|
| 14 |
+
"code_size": 16384,
|
| 15 |
+
"prompt_len": 10,
|
| 16 |
+
"frame_rate": 25,
|
| 17 |
+
"max_position_embeddings": 10000,
|
| 18 |
+
"max_position_embeddings_sub": 10000,
|
| 19 |
+
"rope_theta": 500000.0,
|
| 20 |
+
"rope_theta_sub": 500000.0,
|
| 21 |
+
"cfg_coef": 1.5
|
| 22 |
+
},
|
| 23 |
+
"components": {
|
| 24 |
+
"audiolm": "converted to MLX safetensors",
|
| 25 |
+
"qwen_tokenizer": "copied for lyric/style tokenization",
|
| 26 |
+
"audio_decoder": "use official PyTorch Flow1dVAE/separate tokenizer bridge for now"
|
| 27 |
+
},
|
| 28 |
+
"skipped_keys": [
|
| 29 |
+
"audiolm.out_norm.weight",
|
| 30 |
+
"audiolm.out_norm.bias",
|
| 31 |
+
"audiolm.transformer2.lm_head.weight"
|
| 32 |
+
],
|
| 33 |
+
"weight_files": [
|
| 34 |
+
"model-00001-of-00094.safetensors",
|
| 35 |
+
"model-00002-of-00094.safetensors",
|
| 36 |
+
"model-00003-of-00094.safetensors",
|
| 37 |
+
"model-00004-of-00094.safetensors",
|
| 38 |
+
"model-00005-of-00094.safetensors",
|
| 39 |
+
"model-00006-of-00094.safetensors",
|
| 40 |
+
"model-00007-of-00094.safetensors",
|
| 41 |
+
"model-00008-of-00094.safetensors",
|
| 42 |
+
"model-00009-of-00094.safetensors",
|
| 43 |
+
"model-00010-of-00094.safetensors",
|
| 44 |
+
"model-00011-of-00094.safetensors",
|
| 45 |
+
"model-00012-of-00094.safetensors",
|
| 46 |
+
"model-00013-of-00094.safetensors",
|
| 47 |
+
"model-00014-of-00094.safetensors",
|
| 48 |
+
"model-00015-of-00094.safetensors",
|
| 49 |
+
"model-00016-of-00094.safetensors",
|
| 50 |
+
"model-00017-of-00094.safetensors",
|
| 51 |
+
"model-00018-of-00094.safetensors",
|
| 52 |
+
"model-00019-of-00094.safetensors",
|
| 53 |
+
"model-00020-of-00094.safetensors",
|
| 54 |
+
"model-00021-of-00094.safetensors",
|
| 55 |
+
"model-00022-of-00094.safetensors",
|
| 56 |
+
"model-00023-of-00094.safetensors",
|
| 57 |
+
"model-00024-of-00094.safetensors",
|
| 58 |
+
"model-00025-of-00094.safetensors",
|
| 59 |
+
"model-00026-of-00094.safetensors",
|
| 60 |
+
"model-00027-of-00094.safetensors",
|
| 61 |
+
"model-00028-of-00094.safetensors",
|
| 62 |
+
"model-00029-of-00094.safetensors",
|
| 63 |
+
"model-00030-of-00094.safetensors",
|
| 64 |
+
"model-00031-of-00094.safetensors",
|
| 65 |
+
"model-00032-of-00094.safetensors",
|
| 66 |
+
"model-00033-of-00094.safetensors",
|
| 67 |
+
"model-00034-of-00094.safetensors",
|
| 68 |
+
"model-00035-of-00094.safetensors",
|
| 69 |
+
"model-00036-of-00094.safetensors",
|
| 70 |
+
"model-00037-of-00094.safetensors",
|
| 71 |
+
"model-00038-of-00094.safetensors",
|
| 72 |
+
"model-00039-of-00094.safetensors",
|
| 73 |
+
"model-00040-of-00094.safetensors",
|
| 74 |
+
"model-00041-of-00094.safetensors",
|
| 75 |
+
"model-00042-of-00094.safetensors",
|
| 76 |
+
"model-00043-of-00094.safetensors",
|
| 77 |
+
"model-00044-of-00094.safetensors",
|
| 78 |
+
"model-00045-of-00094.safetensors",
|
| 79 |
+
"model-00046-of-00094.safetensors",
|
| 80 |
+
"model-00047-of-00094.safetensors",
|
| 81 |
+
"model-00048-of-00094.safetensors",
|
| 82 |
+
"model-00049-of-00094.safetensors",
|
| 83 |
+
"model-00050-of-00094.safetensors",
|
| 84 |
+
"model-00051-of-00094.safetensors",
|
| 85 |
+
"model-00052-of-00094.safetensors",
|
| 86 |
+
"model-00053-of-00094.safetensors",
|
| 87 |
+
"model-00054-of-00094.safetensors",
|
| 88 |
+
"model-00055-of-00094.safetensors",
|
| 89 |
+
"model-00056-of-00094.safetensors",
|
| 90 |
+
"model-00057-of-00094.safetensors",
|
| 91 |
+
"model-00058-of-00094.safetensors",
|
| 92 |
+
"model-00059-of-00094.safetensors",
|
| 93 |
+
"model-00060-of-00094.safetensors",
|
| 94 |
+
"model-00061-of-00094.safetensors",
|
| 95 |
+
"model-00062-of-00094.safetensors",
|
| 96 |
+
"model-00063-of-00094.safetensors",
|
| 97 |
+
"model-00064-of-00094.safetensors",
|
| 98 |
+
"model-00065-of-00094.safetensors",
|
| 99 |
+
"model-00066-of-00094.safetensors",
|
| 100 |
+
"model-00067-of-00094.safetensors",
|
| 101 |
+
"model-00068-of-00094.safetensors",
|
| 102 |
+
"model-00069-of-00094.safetensors",
|
| 103 |
+
"model-00070-of-00094.safetensors",
|
| 104 |
+
"model-00071-of-00094.safetensors",
|
| 105 |
+
"model-00072-of-00094.safetensors",
|
| 106 |
+
"model-00073-of-00094.safetensors",
|
| 107 |
+
"model-00074-of-00094.safetensors",
|
| 108 |
+
"model-00075-of-00094.safetensors",
|
| 109 |
+
"model-00076-of-00094.safetensors",
|
| 110 |
+
"model-00077-of-00094.safetensors",
|
| 111 |
+
"model-00078-of-00094.safetensors",
|
| 112 |
+
"model-00079-of-00094.safetensors",
|
| 113 |
+
"model-00080-of-00094.safetensors",
|
| 114 |
+
"model-00081-of-00094.safetensors",
|
| 115 |
+
"model-00082-of-00094.safetensors",
|
| 116 |
+
"model-00083-of-00094.safetensors",
|
| 117 |
+
"model-00084-of-00094.safetensors",
|
| 118 |
+
"model-00085-of-00094.safetensors",
|
| 119 |
+
"model-00086-of-00094.safetensors",
|
| 120 |
+
"model-00087-of-00094.safetensors",
|
| 121 |
+
"model-00088-of-00094.safetensors",
|
| 122 |
+
"model-00089-of-00094.safetensors",
|
| 123 |
+
"model-00090-of-00094.safetensors",
|
| 124 |
+
"model-00091-of-00094.safetensors",
|
| 125 |
+
"model-00092-of-00094.safetensors",
|
| 126 |
+
"model-00093-of-00094.safetensors",
|
| 127 |
+
"model-00094-of-00094.safetensors"
|
| 128 |
+
]
|
| 129 |
+
}
|
config.official.yaml
ADDED
|
@@ -0,0 +1,141 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# ================ Train Config ================ #
|
| 2 |
+
lyric_processor:
|
| 3 |
+
max_dur: 270
|
| 4 |
+
min_dur: 30
|
| 5 |
+
prompt_len: 10
|
| 6 |
+
pad_to_max: true
|
| 7 |
+
|
| 8 |
+
|
| 9 |
+
# ================ Audio tokenzier ================ #
|
| 10 |
+
audio_tokenizer_checkpoint: Flow1dVAE1rvq_./ckpt/model_1rvq/model_2_fixed.safetensors
|
| 11 |
+
audio_tokenizer_frame_rate: 25
|
| 12 |
+
audio_tokenizer_code_depth: 1
|
| 13 |
+
sample_rate: 48000
|
| 14 |
+
|
| 15 |
+
audio_tokenizer_checkpoint_sep: Flow1dVAESeparate_./ckpt/model_septoken/model_2.safetensors
|
| 16 |
+
audio_tokenizer_frame_rate_sep: 25
|
| 17 |
+
audio_tokenizer_code_depth_sep: 2
|
| 18 |
+
sample_rate_sep: 48000
|
| 19 |
+
|
| 20 |
+
# ================ VAE ================ #
|
| 21 |
+
vae_config: ./ckpt/vae/stable_audio_1920_vae.json
|
| 22 |
+
vae_model: ./ckpt/vae/autoencoder_music_1320k.ckpt
|
| 23 |
+
|
| 24 |
+
# ================== LM =========================== #
|
| 25 |
+
lm:
|
| 26 |
+
lm_type: Llama # [Llama]
|
| 27 |
+
dim: 1536
|
| 28 |
+
intermediate_size: 8960
|
| 29 |
+
num_heads: 12
|
| 30 |
+
num_layers: 28
|
| 31 |
+
num_layers_sub: 12
|
| 32 |
+
code_depth: 3
|
| 33 |
+
code_size: 16384
|
| 34 |
+
max_position_embeddings: 10000
|
| 35 |
+
max_position_embeddings_sub: 10000
|
| 36 |
+
rope_theta: 500000.0
|
| 37 |
+
rope_theta_sub: 500000.0
|
| 38 |
+
dropout: 0.0
|
| 39 |
+
use_flash_attn_2: true
|
| 40 |
+
activation: gelu
|
| 41 |
+
norm_first: true
|
| 42 |
+
bias_ff: false
|
| 43 |
+
bias_attn: false
|
| 44 |
+
causal: true
|
| 45 |
+
custom: false
|
| 46 |
+
memory_efficient: true
|
| 47 |
+
attention_as_float32: false
|
| 48 |
+
layer_scale: null
|
| 49 |
+
positional_embedding: sin
|
| 50 |
+
xpos: false
|
| 51 |
+
checkpointing: torch
|
| 52 |
+
weight_init: gaussian
|
| 53 |
+
depthwise_init: current
|
| 54 |
+
zero_bias_init: true
|
| 55 |
+
norm: layer_norm
|
| 56 |
+
cross_attention: false
|
| 57 |
+
qk_layer_norm: false
|
| 58 |
+
qk_layer_norm_cross: false
|
| 59 |
+
attention_dropout: null
|
| 60 |
+
kv_repeat: 1
|
| 61 |
+
|
| 62 |
+
codebooks_pattern:
|
| 63 |
+
modeling: delay
|
| 64 |
+
delay:
|
| 65 |
+
delays: [ 0, 250, 250 ]
|
| 66 |
+
flatten_first: 0
|
| 67 |
+
empty_initial: 0
|
| 68 |
+
|
| 69 |
+
# ================ Conditioners ===================== #
|
| 70 |
+
classifier_free_guidance:
|
| 71 |
+
# drop all conditions simultaneously
|
| 72 |
+
training_dropout: 0.15
|
| 73 |
+
inference_coef: 1.5
|
| 74 |
+
|
| 75 |
+
attribute_dropout:
|
| 76 |
+
# drop each condition separately
|
| 77 |
+
args:
|
| 78 |
+
active_on_eval: false
|
| 79 |
+
text:
|
| 80 |
+
description: 0.0
|
| 81 |
+
type_info: 0.2
|
| 82 |
+
audio:
|
| 83 |
+
prompt_audio: 0.5
|
| 84 |
+
|
| 85 |
+
|
| 86 |
+
use_text_training: True
|
| 87 |
+
fuser:
|
| 88 |
+
sum: []
|
| 89 |
+
prepend: [ description, prompt_audio, type_info ] # this order is the SAME with the input concatenation order
|
| 90 |
+
|
| 91 |
+
conditioners:
|
| 92 |
+
prompt_audio:
|
| 93 |
+
model: qt_embedding
|
| 94 |
+
qt_embedding:
|
| 95 |
+
code_size: 16384
|
| 96 |
+
code_depth: 3
|
| 97 |
+
max_len: ${eval:${prompt_len}*${audio_tokenizer_frame_rate}+2} # 25*10+2+1
|
| 98 |
+
description:
|
| 99 |
+
model: QwTokenizer
|
| 100 |
+
QwTokenizer:
|
| 101 |
+
token_path: third_party/Qwen2-7B
|
| 102 |
+
max_len: 600
|
| 103 |
+
add_token_list: ${load_yaml:conf/vocab.yaml}
|
| 104 |
+
type_info:
|
| 105 |
+
model: QwTextTokenizer
|
| 106 |
+
QwTextTokenizer:
|
| 107 |
+
token_path: third_party/Qwen2-7B
|
| 108 |
+
max_len: 100
|
| 109 |
+
|
| 110 |
+
offload:
|
| 111 |
+
audiolm:
|
| 112 |
+
offload_module: self
|
| 113 |
+
cpu_mem_gb: 0
|
| 114 |
+
pre_copy_step: 1
|
| 115 |
+
clean_cache_after_forward: false
|
| 116 |
+
dtype: torch.float16
|
| 117 |
+
offload_layer_dict:
|
| 118 |
+
transformer: 4
|
| 119 |
+
transformer2: 4
|
| 120 |
+
ignore_layer_list: []
|
| 121 |
+
clean_cache_wrapper:
|
| 122 |
+
module: self
|
| 123 |
+
method_name: _sample_next_token
|
| 124 |
+
diff_mem_gb_thre: 2
|
| 125 |
+
debug: false
|
| 126 |
+
|
| 127 |
+
wav_tokenizer_diffusion:
|
| 128 |
+
offload_module: self.model.model
|
| 129 |
+
pre_copy_step: 1
|
| 130 |
+
clean_cache_after_forward: false
|
| 131 |
+
cpu_mem_gb: -1
|
| 132 |
+
dtype: null
|
| 133 |
+
offload_layer_dict:
|
| 134 |
+
cfm_wrapper: 5
|
| 135 |
+
hubert: 4
|
| 136 |
+
ignore_layer_list: []
|
| 137 |
+
clean_cache_wrapper:
|
| 138 |
+
module: self.model.model.cfm_wrapper.estimator
|
| 139 |
+
method_name: forward
|
| 140 |
+
diff_mem_gb_thre: 1
|
| 141 |
+
debug: false
|
mlx_manifest.json
ADDED
|
@@ -0,0 +1,129 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"format": "songgeneration-mlx-audiolm",
|
| 3 |
+
"source": "tencent/SongGeneration",
|
| 4 |
+
"official_code": "https://github.com/tencent-ailab/songgeneration",
|
| 5 |
+
"variant": "v2-medium",
|
| 6 |
+
"precision": "bfloat16",
|
| 7 |
+
"runtime": {
|
| 8 |
+
"hidden_size": 1536,
|
| 9 |
+
"intermediate_size": 8960,
|
| 10 |
+
"num_heads": 12,
|
| 11 |
+
"num_layers": 28,
|
| 12 |
+
"num_layers_sub": 12,
|
| 13 |
+
"code_depth": 3,
|
| 14 |
+
"code_size": 16384,
|
| 15 |
+
"prompt_len": 10,
|
| 16 |
+
"frame_rate": 25,
|
| 17 |
+
"max_position_embeddings": 10000,
|
| 18 |
+
"max_position_embeddings_sub": 10000,
|
| 19 |
+
"rope_theta": 500000.0,
|
| 20 |
+
"rope_theta_sub": 500000.0,
|
| 21 |
+
"cfg_coef": 1.5
|
| 22 |
+
},
|
| 23 |
+
"components": {
|
| 24 |
+
"audiolm": "converted to MLX safetensors",
|
| 25 |
+
"qwen_tokenizer": "copied for lyric/style tokenization",
|
| 26 |
+
"audio_decoder": "use official PyTorch Flow1dVAE/separate tokenizer bridge for now"
|
| 27 |
+
},
|
| 28 |
+
"skipped_keys": [
|
| 29 |
+
"audiolm.out_norm.weight",
|
| 30 |
+
"audiolm.out_norm.bias",
|
| 31 |
+
"audiolm.transformer2.lm_head.weight"
|
| 32 |
+
],
|
| 33 |
+
"weight_files": [
|
| 34 |
+
"model-00001-of-00094.safetensors",
|
| 35 |
+
"model-00002-of-00094.safetensors",
|
| 36 |
+
"model-00003-of-00094.safetensors",
|
| 37 |
+
"model-00004-of-00094.safetensors",
|
| 38 |
+
"model-00005-of-00094.safetensors",
|
| 39 |
+
"model-00006-of-00094.safetensors",
|
| 40 |
+
"model-00007-of-00094.safetensors",
|
| 41 |
+
"model-00008-of-00094.safetensors",
|
| 42 |
+
"model-00009-of-00094.safetensors",
|
| 43 |
+
"model-00010-of-00094.safetensors",
|
| 44 |
+
"model-00011-of-00094.safetensors",
|
| 45 |
+
"model-00012-of-00094.safetensors",
|
| 46 |
+
"model-00013-of-00094.safetensors",
|
| 47 |
+
"model-00014-of-00094.safetensors",
|
| 48 |
+
"model-00015-of-00094.safetensors",
|
| 49 |
+
"model-00016-of-00094.safetensors",
|
| 50 |
+
"model-00017-of-00094.safetensors",
|
| 51 |
+
"model-00018-of-00094.safetensors",
|
| 52 |
+
"model-00019-of-00094.safetensors",
|
| 53 |
+
"model-00020-of-00094.safetensors",
|
| 54 |
+
"model-00021-of-00094.safetensors",
|
| 55 |
+
"model-00022-of-00094.safetensors",
|
| 56 |
+
"model-00023-of-00094.safetensors",
|
| 57 |
+
"model-00024-of-00094.safetensors",
|
| 58 |
+
"model-00025-of-00094.safetensors",
|
| 59 |
+
"model-00026-of-00094.safetensors",
|
| 60 |
+
"model-00027-of-00094.safetensors",
|
| 61 |
+
"model-00028-of-00094.safetensors",
|
| 62 |
+
"model-00029-of-00094.safetensors",
|
| 63 |
+
"model-00030-of-00094.safetensors",
|
| 64 |
+
"model-00031-of-00094.safetensors",
|
| 65 |
+
"model-00032-of-00094.safetensors",
|
| 66 |
+
"model-00033-of-00094.safetensors",
|
| 67 |
+
"model-00034-of-00094.safetensors",
|
| 68 |
+
"model-00035-of-00094.safetensors",
|
| 69 |
+
"model-00036-of-00094.safetensors",
|
| 70 |
+
"model-00037-of-00094.safetensors",
|
| 71 |
+
"model-00038-of-00094.safetensors",
|
| 72 |
+
"model-00039-of-00094.safetensors",
|
| 73 |
+
"model-00040-of-00094.safetensors",
|
| 74 |
+
"model-00041-of-00094.safetensors",
|
| 75 |
+
"model-00042-of-00094.safetensors",
|
| 76 |
+
"model-00043-of-00094.safetensors",
|
| 77 |
+
"model-00044-of-00094.safetensors",
|
| 78 |
+
"model-00045-of-00094.safetensors",
|
| 79 |
+
"model-00046-of-00094.safetensors",
|
| 80 |
+
"model-00047-of-00094.safetensors",
|
| 81 |
+
"model-00048-of-00094.safetensors",
|
| 82 |
+
"model-00049-of-00094.safetensors",
|
| 83 |
+
"model-00050-of-00094.safetensors",
|
| 84 |
+
"model-00051-of-00094.safetensors",
|
| 85 |
+
"model-00052-of-00094.safetensors",
|
| 86 |
+
"model-00053-of-00094.safetensors",
|
| 87 |
+
"model-00054-of-00094.safetensors",
|
| 88 |
+
"model-00055-of-00094.safetensors",
|
| 89 |
+
"model-00056-of-00094.safetensors",
|
| 90 |
+
"model-00057-of-00094.safetensors",
|
| 91 |
+
"model-00058-of-00094.safetensors",
|
| 92 |
+
"model-00059-of-00094.safetensors",
|
| 93 |
+
"model-00060-of-00094.safetensors",
|
| 94 |
+
"model-00061-of-00094.safetensors",
|
| 95 |
+
"model-00062-of-00094.safetensors",
|
| 96 |
+
"model-00063-of-00094.safetensors",
|
| 97 |
+
"model-00064-of-00094.safetensors",
|
| 98 |
+
"model-00065-of-00094.safetensors",
|
| 99 |
+
"model-00066-of-00094.safetensors",
|
| 100 |
+
"model-00067-of-00094.safetensors",
|
| 101 |
+
"model-00068-of-00094.safetensors",
|
| 102 |
+
"model-00069-of-00094.safetensors",
|
| 103 |
+
"model-00070-of-00094.safetensors",
|
| 104 |
+
"model-00071-of-00094.safetensors",
|
| 105 |
+
"model-00072-of-00094.safetensors",
|
| 106 |
+
"model-00073-of-00094.safetensors",
|
| 107 |
+
"model-00074-of-00094.safetensors",
|
| 108 |
+
"model-00075-of-00094.safetensors",
|
| 109 |
+
"model-00076-of-00094.safetensors",
|
| 110 |
+
"model-00077-of-00094.safetensors",
|
| 111 |
+
"model-00078-of-00094.safetensors",
|
| 112 |
+
"model-00079-of-00094.safetensors",
|
| 113 |
+
"model-00080-of-00094.safetensors",
|
| 114 |
+
"model-00081-of-00094.safetensors",
|
| 115 |
+
"model-00082-of-00094.safetensors",
|
| 116 |
+
"model-00083-of-00094.safetensors",
|
| 117 |
+
"model-00084-of-00094.safetensors",
|
| 118 |
+
"model-00085-of-00094.safetensors",
|
| 119 |
+
"model-00086-of-00094.safetensors",
|
| 120 |
+
"model-00087-of-00094.safetensors",
|
| 121 |
+
"model-00088-of-00094.safetensors",
|
| 122 |
+
"model-00089-of-00094.safetensors",
|
| 123 |
+
"model-00090-of-00094.safetensors",
|
| 124 |
+
"model-00091-of-00094.safetensors",
|
| 125 |
+
"model-00092-of-00094.safetensors",
|
| 126 |
+
"model-00093-of-00094.safetensors",
|
| 127 |
+
"model-00094-of-00094.safetensors"
|
| 128 |
+
]
|
| 129 |
+
}
|
model-00002-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6cadfb8ae74b6d48e4ebc2e5788220ef00a70618d6ba90aef8b3a0722605d50b
|
| 3 |
+
size 50952384
|
model-00003-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dcb52f514436ea6ffc20dc387de60b4add69f57cec09116601b139d64340ec43
|
| 3 |
+
size 50337888
|
model-00006-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d4ed1114e824b085cc553c9683e170681abb37ca8535d45e5951d9d3a2f9cffb
|
| 3 |
+
size 50334816
|
model-00007-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b9911d98ef55c85f7aa2c3d2718894e483beac43219e4e90c413c76dbcea5e60
|
| 3 |
+
size 64497096
|
model-00008-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:951917930b5bd938de09c6429eee274e934cfa223b140fe8446e75be83b2733f
|
| 3 |
+
size 50337896
|
model-00009-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a4597fa6e467841a9a6f63c15480161f177cd750c59cd42a1141fcc706c97dc8
|
| 3 |
+
size 50337896
|
model-00011-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:61ad33d538807175ed0832a8282142919417168155cb56b828d326419a2acf12
|
| 3 |
+
size 50338016
|
model-00014-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6e45a51be1539e727ac4a858015c4385add0b303670183b88ac109af5f9a9f78
|
| 3 |
+
size 55050480
|
model-00015-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1c726f524b65d9cc1a07c526ab20168a04f4c36a1a1524d6b113900fcfedfbc4
|
| 3 |
+
size 46406464
|
model-00022-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:42edc0780be172b3f4c84836109be0c9a7ecc2994af0641fdb2f50f2cfbd5562
|
| 3 |
+
size 55050480
|
model-00023-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bf8dd4958d468fe8709ab405b9c07d12b7fd299f71fe485b54ea060ea927cfa8
|
| 3 |
+
size 46406472
|
model-00026-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:74aff76e04fae7b6c1589f2f7c5d3e965537f6c3df892374fa2610f148a6101a
|
| 3 |
+
size 55050480
|
model-00027-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:146473447f762582cca32cdc055a060dc16cefe8e49624769df2f23e1a208837
|
| 3 |
+
size 46406472
|
model-00030-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a2c8da44adb3faebaeb402d02b498761a86709a75437554febf2eb968a863727
|
| 3 |
+
size 55050480
|
model-00031-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e7d930a844e898a1a78bd3c8a00218a4d9662e0e9da45b4849c9a19c56eda045
|
| 3 |
+
size 46406472
|
model-00034-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5c93107da2de073d647e5275a173b39b3fb9f0a97b5e6c04ba43cf682d024f7a
|
| 3 |
+
size 55050480
|
model-00035-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0ebcbb5b1b0a4fe49fd68bdb2c9f3c5013fae802f3ddd4e1aabc96bdb6a2636a
|
| 3 |
+
size 46406472
|
model-00042-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:899df93794103faccfb0a821c3b78ca13df222c88e2bf33adfeadda5330ebfba
|
| 3 |
+
size 55050480
|
model-00043-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:07edf0fcfb7538e2cc1a22cfafe2437480e8a6d3bcb7cf04244df3a437aed1ff
|
| 3 |
+
size 46406472
|
model-00046-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e451c621dd5db114a455e9b6e9dc305a91b5f39a167624badcddae493612b4aa
|
| 3 |
+
size 55050480
|
model-00047-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf3ec8b6dcfa6e6a6b231d7f97274efbfc59eb9a7602d117559f406f07100683
|
| 3 |
+
size 46406472
|
model-00048-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7c08a9fc1e546eb3434655ba908f4e3b17999087652bc2c22a9cb7ee309be92f
|
| 3 |
+
size 55050480
|
model-00049-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c34aedf0ed283af53e0554dcd40ed19e289a60ead254c42f845b440cf9b9f630
|
| 3 |
+
size 46406472
|
model-00050-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c50db817129b34b060e598dbeec29acb0ceb8dd9d7382ea7bd7fe5203829fae
|
| 3 |
+
size 55050480
|
model-00054-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bf4ae42b4b0e70eff41bd77bb66a027d84aa60403ff1b3869628dcc7ed9dd9d5
|
| 3 |
+
size 55050480
|
model-00055-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f029578a882d2c4914b6055f5449a00e6707b23d93d0d2cfd2eb5e3a0471002b
|
| 3 |
+
size 46406464
|
model-00062-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3ca1d5e3b97e32370e3fae8fdd6092e47323e0ffcc80e41b07ea79e51006469c
|
| 3 |
+
size 55050480
|
model-00063-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:754629ae6c6443bd892f021eb27705aed39cd354c1185dfc7f2e46aab5ac3944
|
| 3 |
+
size 46406464
|
model-00066-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5f34e5b8e11c61b2a9e47b89dce969b87d8f00a3cb24d545b8beec82c5db888b
|
| 3 |
+
size 55050480
|
model-00067-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:123a71fd9dba5ecb7fc137f2931875698d1de1bc465cb5b5a372eddc84a977ed
|
| 3 |
+
size 46406440
|
model-00068-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9684f179ececc1eae638adf53e0f76983f677ad4ec6764c5763497b33f43b4ef
|
| 3 |
+
size 50338016
|
model-00070-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ed88ac4f04e8dea7bf2cd41479135467f52c45a9a239055f66570abb6417441
|
| 3 |
+
size 46406472
|
model-00071-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3315fbc19510741c9bd94adcc7df01883804f7be53725cc402a101a687d51b8a
|
| 3 |
+
size 55050480
|
model-00074-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9110d4f8bae6fe2da7f9ad9ec154e5b1bf09ebaca197b079d6c81d20b0b4b3f1
|
| 3 |
+
size 46406480
|
model-00075-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06a14191c06c1266b2788e3660083191b57f2b14d63076acfb42e8dfbfc26623
|
| 3 |
+
size 55050480
|
model-00082-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:969c6b5142eea9711e46df6e194d9c1980fe4681910d739e209727e61b609cd8
|
| 3 |
+
size 46406472
|
model-00083-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:50109b89f3860f3251c1d76df8e2dc0959d2f4f0a9f282f6ce5a7ae99f3dd843
|
| 3 |
+
size 55050480
|
model-00086-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5dbf6c14cb3dc91039c2a36fe686a77850ff61336325b606444566a64c239ae6
|
| 3 |
+
size 46406472
|
model-00087-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0272083f55838ce55573dedcc36a95ade5dd1b14bab48e13ad4eb53c138e1a81
|
| 3 |
+
size 55050480
|
model-00088-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:99f143e2f433f409e011f2b2e7d250c7fd9f5768a88e397d99b05a9fd1c6b9ca
|
| 3 |
+
size 46406472
|
model-00089-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:84b54530ac37d9bd45ffee6a2a50ac82d6fa68de8efc6352b202e7a6745960cb
|
| 3 |
+
size 55050480
|
model-00090-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0c0d504cfb768681ce8560804a9e3f91d62dd1d10982aa8a7da9905726df95fc
|
| 3 |
+
size 46406472
|
model-00091-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:36c04cf95c9a1275056fe97c410b1d29e4fdf2ca5522c9cc5ac21f4dd6645805
|
| 3 |
+
size 55050480
|
model-00094-of-00094.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8a1e5e0712573d51ee2c96c4c92575ffc76a7a5a5396e6d43e1a2bbaad4957f9
|
| 3 |
+
size 465875056
|
model.safetensors.index.json
ADDED
|
@@ -0,0 +1,390 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metadata": {
|
| 3 |
+
"total_size": 5608833024
|
| 4 |
+
},
|
| 5 |
+
"weight_map": {
|
| 6 |
+
"description_output_proj.weight": "model-00001-of-00094.safetensors",
|
| 7 |
+
"description_structure_emb.weight": "model-00002-of-00094.safetensors",
|
| 8 |
+
"emb0.weight": "model-00002-of-00094.safetensors",
|
| 9 |
+
"layer2_emb0.weight": "model-00003-of-00094.safetensors",
|
| 10 |
+
"layer2_emb1.weight": "model-00004-of-00094.safetensors",
|
| 11 |
+
"layer2_emb2.weight": "model-00005-of-00094.safetensors",
|
| 12 |
+
"linears0.weight": "model-00006-of-00094.safetensors",
|
| 13 |
+
"linears1.weight": "model-00007-of-00094.safetensors",
|
| 14 |
+
"mlp.linear0.bias": "model-00007-of-00094.safetensors",
|
| 15 |
+
"mlp.linear0.weight": "model-00007-of-00094.safetensors",
|
| 16 |
+
"mlp.linear2.bias": "model-00007-of-00094.safetensors",
|
| 17 |
+
"mlp.linear2.weight": "model-00007-of-00094.safetensors",
|
| 18 |
+
"prompt_audio_emb0.weight": "model-00008-of-00094.safetensors",
|
| 19 |
+
"prompt_audio_emb1.weight": "model-00009-of-00094.safetensors",
|
| 20 |
+
"prompt_audio_emb2.weight": "model-00010-of-00094.safetensors",
|
| 21 |
+
"prompt_audio_eot": "model-00010-of-00094.safetensors",
|
| 22 |
+
"prompt_audio_layer2_eot": "model-00010-of-00094.safetensors",
|
| 23 |
+
"transformer.embed_tokens.weight": "model-00011-of-00094.safetensors",
|
| 24 |
+
"transformer.layers.0.input_layernorm.weight": "model-00011-of-00094.safetensors",
|
| 25 |
+
"transformer.layers.0.mlp.down_proj.weight": "model-00012-of-00094.safetensors",
|
| 26 |
+
"transformer.layers.0.mlp.gate_proj.weight": "model-00012-of-00094.safetensors",
|
| 27 |
+
"transformer.layers.0.mlp.up_proj.weight": "model-00013-of-00094.safetensors",
|
| 28 |
+
"transformer.layers.0.post_attention_layernorm.weight": "model-00013-of-00094.safetensors",
|
| 29 |
+
"transformer.layers.0.self_attn.k_proj.weight": "model-00013-of-00094.safetensors",
|
| 30 |
+
"transformer.layers.0.self_attn.o_proj.weight": "model-00013-of-00094.safetensors",
|
| 31 |
+
"transformer.layers.0.self_attn.q_proj.weight": "model-00013-of-00094.safetensors",
|
| 32 |
+
"transformer.layers.0.self_attn.v_proj.weight": "model-00013-of-00094.safetensors",
|
| 33 |
+
"transformer.layers.1.input_layernorm.weight": "model-00013-of-00094.safetensors",
|
| 34 |
+
"transformer.layers.1.mlp.down_proj.weight": "model-00014-of-00094.safetensors",
|
| 35 |
+
"transformer.layers.1.mlp.gate_proj.weight": "model-00014-of-00094.safetensors",
|
| 36 |
+
"transformer.layers.1.mlp.up_proj.weight": "model-00015-of-00094.safetensors",
|
| 37 |
+
"transformer.layers.1.post_attention_layernorm.weight": "model-00015-of-00094.safetensors",
|
| 38 |
+
"transformer.layers.1.self_attn.k_proj.weight": "model-00015-of-00094.safetensors",
|
| 39 |
+
"transformer.layers.1.self_attn.o_proj.weight": "model-00015-of-00094.safetensors",
|
| 40 |
+
"transformer.layers.1.self_attn.q_proj.weight": "model-00015-of-00094.safetensors",
|
| 41 |
+
"transformer.layers.1.self_attn.v_proj.weight": "model-00015-of-00094.safetensors",
|
| 42 |
+
"transformer.layers.10.input_layernorm.weight": "model-00015-of-00094.safetensors",
|
| 43 |
+
"transformer.layers.10.mlp.down_proj.weight": "model-00016-of-00094.safetensors",
|
| 44 |
+
"transformer.layers.10.mlp.gate_proj.weight": "model-00016-of-00094.safetensors",
|
| 45 |
+
"transformer.layers.10.mlp.up_proj.weight": "model-00017-of-00094.safetensors",
|
| 46 |
+
"transformer.layers.10.post_attention_layernorm.weight": "model-00017-of-00094.safetensors",
|
| 47 |
+
"transformer.layers.10.self_attn.k_proj.weight": "model-00017-of-00094.safetensors",
|
| 48 |
+
"transformer.layers.10.self_attn.o_proj.weight": "model-00017-of-00094.safetensors",
|
| 49 |
+
"transformer.layers.10.self_attn.q_proj.weight": "model-00017-of-00094.safetensors",
|
| 50 |
+
"transformer.layers.10.self_attn.v_proj.weight": "model-00017-of-00094.safetensors",
|
| 51 |
+
"transformer.layers.11.input_layernorm.weight": "model-00017-of-00094.safetensors",
|
| 52 |
+
"transformer.layers.11.mlp.down_proj.weight": "model-00018-of-00094.safetensors",
|
| 53 |
+
"transformer.layers.11.mlp.gate_proj.weight": "model-00018-of-00094.safetensors",
|
| 54 |
+
"transformer.layers.11.mlp.up_proj.weight": "model-00019-of-00094.safetensors",
|
| 55 |
+
"transformer.layers.11.post_attention_layernorm.weight": "model-00019-of-00094.safetensors",
|
| 56 |
+
"transformer.layers.11.self_attn.k_proj.weight": "model-00019-of-00094.safetensors",
|
| 57 |
+
"transformer.layers.11.self_attn.o_proj.weight": "model-00019-of-00094.safetensors",
|
| 58 |
+
"transformer.layers.11.self_attn.q_proj.weight": "model-00019-of-00094.safetensors",
|
| 59 |
+
"transformer.layers.11.self_attn.v_proj.weight": "model-00019-of-00094.safetensors",
|
| 60 |
+
"transformer.layers.12.input_layernorm.weight": "model-00019-of-00094.safetensors",
|
| 61 |
+
"transformer.layers.12.mlp.down_proj.weight": "model-00020-of-00094.safetensors",
|
| 62 |
+
"transformer.layers.12.mlp.gate_proj.weight": "model-00020-of-00094.safetensors",
|
| 63 |
+
"transformer.layers.12.mlp.up_proj.weight": "model-00021-of-00094.safetensors",
|
| 64 |
+
"transformer.layers.12.post_attention_layernorm.weight": "model-00021-of-00094.safetensors",
|
| 65 |
+
"transformer.layers.12.self_attn.k_proj.weight": "model-00021-of-00094.safetensors",
|
| 66 |
+
"transformer.layers.12.self_attn.o_proj.weight": "model-00021-of-00094.safetensors",
|
| 67 |
+
"transformer.layers.12.self_attn.q_proj.weight": "model-00021-of-00094.safetensors",
|
| 68 |
+
"transformer.layers.12.self_attn.v_proj.weight": "model-00021-of-00094.safetensors",
|
| 69 |
+
"transformer.layers.13.input_layernorm.weight": "model-00021-of-00094.safetensors",
|
| 70 |
+
"transformer.layers.13.mlp.down_proj.weight": "model-00022-of-00094.safetensors",
|
| 71 |
+
"transformer.layers.13.mlp.gate_proj.weight": "model-00022-of-00094.safetensors",
|
| 72 |
+
"transformer.layers.13.mlp.up_proj.weight": "model-00023-of-00094.safetensors",
|
| 73 |
+
"transformer.layers.13.post_attention_layernorm.weight": "model-00023-of-00094.safetensors",
|
| 74 |
+
"transformer.layers.13.self_attn.k_proj.weight": "model-00023-of-00094.safetensors",
|
| 75 |
+
"transformer.layers.13.self_attn.o_proj.weight": "model-00023-of-00094.safetensors",
|
| 76 |
+
"transformer.layers.13.self_attn.q_proj.weight": "model-00023-of-00094.safetensors",
|
| 77 |
+
"transformer.layers.13.self_attn.v_proj.weight": "model-00023-of-00094.safetensors",
|
| 78 |
+
"transformer.layers.14.input_layernorm.weight": "model-00023-of-00094.safetensors",
|
| 79 |
+
"transformer.layers.14.mlp.down_proj.weight": "model-00024-of-00094.safetensors",
|
| 80 |
+
"transformer.layers.14.mlp.gate_proj.weight": "model-00024-of-00094.safetensors",
|
| 81 |
+
"transformer.layers.14.mlp.up_proj.weight": "model-00025-of-00094.safetensors",
|
| 82 |
+
"transformer.layers.14.post_attention_layernorm.weight": "model-00025-of-00094.safetensors",
|
| 83 |
+
"transformer.layers.14.self_attn.k_proj.weight": "model-00025-of-00094.safetensors",
|
| 84 |
+
"transformer.layers.14.self_attn.o_proj.weight": "model-00025-of-00094.safetensors",
|
| 85 |
+
"transformer.layers.14.self_attn.q_proj.weight": "model-00025-of-00094.safetensors",
|
| 86 |
+
"transformer.layers.14.self_attn.v_proj.weight": "model-00025-of-00094.safetensors",
|
| 87 |
+
"transformer.layers.15.input_layernorm.weight": "model-00025-of-00094.safetensors",
|
| 88 |
+
"transformer.layers.15.mlp.down_proj.weight": "model-00026-of-00094.safetensors",
|
| 89 |
+
"transformer.layers.15.mlp.gate_proj.weight": "model-00026-of-00094.safetensors",
|
| 90 |
+
"transformer.layers.15.mlp.up_proj.weight": "model-00027-of-00094.safetensors",
|
| 91 |
+
"transformer.layers.15.post_attention_layernorm.weight": "model-00027-of-00094.safetensors",
|
| 92 |
+
"transformer.layers.15.self_attn.k_proj.weight": "model-00027-of-00094.safetensors",
|
| 93 |
+
"transformer.layers.15.self_attn.o_proj.weight": "model-00027-of-00094.safetensors",
|
| 94 |
+
"transformer.layers.15.self_attn.q_proj.weight": "model-00027-of-00094.safetensors",
|
| 95 |
+
"transformer.layers.15.self_attn.v_proj.weight": "model-00027-of-00094.safetensors",
|
| 96 |
+
"transformer.layers.16.input_layernorm.weight": "model-00027-of-00094.safetensors",
|
| 97 |
+
"transformer.layers.16.mlp.down_proj.weight": "model-00028-of-00094.safetensors",
|
| 98 |
+
"transformer.layers.16.mlp.gate_proj.weight": "model-00028-of-00094.safetensors",
|
| 99 |
+
"transformer.layers.16.mlp.up_proj.weight": "model-00029-of-00094.safetensors",
|
| 100 |
+
"transformer.layers.16.post_attention_layernorm.weight": "model-00029-of-00094.safetensors",
|
| 101 |
+
"transformer.layers.16.self_attn.k_proj.weight": "model-00029-of-00094.safetensors",
|
| 102 |
+
"transformer.layers.16.self_attn.o_proj.weight": "model-00029-of-00094.safetensors",
|
| 103 |
+
"transformer.layers.16.self_attn.q_proj.weight": "model-00029-of-00094.safetensors",
|
| 104 |
+
"transformer.layers.16.self_attn.v_proj.weight": "model-00029-of-00094.safetensors",
|
| 105 |
+
"transformer.layers.17.input_layernorm.weight": "model-00029-of-00094.safetensors",
|
| 106 |
+
"transformer.layers.17.mlp.down_proj.weight": "model-00030-of-00094.safetensors",
|
| 107 |
+
"transformer.layers.17.mlp.gate_proj.weight": "model-00030-of-00094.safetensors",
|
| 108 |
+
"transformer.layers.17.mlp.up_proj.weight": "model-00031-of-00094.safetensors",
|
| 109 |
+
"transformer.layers.17.post_attention_layernorm.weight": "model-00031-of-00094.safetensors",
|
| 110 |
+
"transformer.layers.17.self_attn.k_proj.weight": "model-00031-of-00094.safetensors",
|
| 111 |
+
"transformer.layers.17.self_attn.o_proj.weight": "model-00031-of-00094.safetensors",
|
| 112 |
+
"transformer.layers.17.self_attn.q_proj.weight": "model-00031-of-00094.safetensors",
|
| 113 |
+
"transformer.layers.17.self_attn.v_proj.weight": "model-00031-of-00094.safetensors",
|
| 114 |
+
"transformer.layers.18.input_layernorm.weight": "model-00031-of-00094.safetensors",
|
| 115 |
+
"transformer.layers.18.mlp.down_proj.weight": "model-00032-of-00094.safetensors",
|
| 116 |
+
"transformer.layers.18.mlp.gate_proj.weight": "model-00032-of-00094.safetensors",
|
| 117 |
+
"transformer.layers.18.mlp.up_proj.weight": "model-00033-of-00094.safetensors",
|
| 118 |
+
"transformer.layers.18.post_attention_layernorm.weight": "model-00033-of-00094.safetensors",
|
| 119 |
+
"transformer.layers.18.self_attn.k_proj.weight": "model-00033-of-00094.safetensors",
|
| 120 |
+
"transformer.layers.18.self_attn.o_proj.weight": "model-00033-of-00094.safetensors",
|
| 121 |
+
"transformer.layers.18.self_attn.q_proj.weight": "model-00033-of-00094.safetensors",
|
| 122 |
+
"transformer.layers.18.self_attn.v_proj.weight": "model-00033-of-00094.safetensors",
|
| 123 |
+
"transformer.layers.19.input_layernorm.weight": "model-00033-of-00094.safetensors",
|
| 124 |
+
"transformer.layers.19.mlp.down_proj.weight": "model-00034-of-00094.safetensors",
|
| 125 |
+
"transformer.layers.19.mlp.gate_proj.weight": "model-00034-of-00094.safetensors",
|
| 126 |
+
"transformer.layers.19.mlp.up_proj.weight": "model-00035-of-00094.safetensors",
|
| 127 |
+
"transformer.layers.19.post_attention_layernorm.weight": "model-00035-of-00094.safetensors",
|
| 128 |
+
"transformer.layers.19.self_attn.k_proj.weight": "model-00035-of-00094.safetensors",
|
| 129 |
+
"transformer.layers.19.self_attn.o_proj.weight": "model-00035-of-00094.safetensors",
|
| 130 |
+
"transformer.layers.19.self_attn.q_proj.weight": "model-00035-of-00094.safetensors",
|
| 131 |
+
"transformer.layers.19.self_attn.v_proj.weight": "model-00035-of-00094.safetensors",
|
| 132 |
+
"transformer.layers.2.input_layernorm.weight": "model-00035-of-00094.safetensors",
|
| 133 |
+
"transformer.layers.2.mlp.down_proj.weight": "model-00036-of-00094.safetensors",
|
| 134 |
+
"transformer.layers.2.mlp.gate_proj.weight": "model-00036-of-00094.safetensors",
|
| 135 |
+
"transformer.layers.2.mlp.up_proj.weight": "model-00037-of-00094.safetensors",
|
| 136 |
+
"transformer.layers.2.post_attention_layernorm.weight": "model-00037-of-00094.safetensors",
|
| 137 |
+
"transformer.layers.2.self_attn.k_proj.weight": "model-00037-of-00094.safetensors",
|
| 138 |
+
"transformer.layers.2.self_attn.o_proj.weight": "model-00037-of-00094.safetensors",
|
| 139 |
+
"transformer.layers.2.self_attn.q_proj.weight": "model-00037-of-00094.safetensors",
|
| 140 |
+
"transformer.layers.2.self_attn.v_proj.weight": "model-00037-of-00094.safetensors",
|
| 141 |
+
"transformer.layers.20.input_layernorm.weight": "model-00037-of-00094.safetensors",
|
| 142 |
+
"transformer.layers.20.mlp.down_proj.weight": "model-00038-of-00094.safetensors",
|
| 143 |
+
"transformer.layers.20.mlp.gate_proj.weight": "model-00038-of-00094.safetensors",
|
| 144 |
+
"transformer.layers.20.mlp.up_proj.weight": "model-00039-of-00094.safetensors",
|
| 145 |
+
"transformer.layers.20.post_attention_layernorm.weight": "model-00039-of-00094.safetensors",
|
| 146 |
+
"transformer.layers.20.self_attn.k_proj.weight": "model-00039-of-00094.safetensors",
|
| 147 |
+
"transformer.layers.20.self_attn.o_proj.weight": "model-00039-of-00094.safetensors",
|
| 148 |
+
"transformer.layers.20.self_attn.q_proj.weight": "model-00039-of-00094.safetensors",
|
| 149 |
+
"transformer.layers.20.self_attn.v_proj.weight": "model-00039-of-00094.safetensors",
|
| 150 |
+
"transformer.layers.21.input_layernorm.weight": "model-00039-of-00094.safetensors",
|
| 151 |
+
"transformer.layers.21.mlp.down_proj.weight": "model-00040-of-00094.safetensors",
|
| 152 |
+
"transformer.layers.21.mlp.gate_proj.weight": "model-00040-of-00094.safetensors",
|
| 153 |
+
"transformer.layers.21.mlp.up_proj.weight": "model-00041-of-00094.safetensors",
|
| 154 |
+
"transformer.layers.21.post_attention_layernorm.weight": "model-00041-of-00094.safetensors",
|
| 155 |
+
"transformer.layers.21.self_attn.k_proj.weight": "model-00041-of-00094.safetensors",
|
| 156 |
+
"transformer.layers.21.self_attn.o_proj.weight": "model-00041-of-00094.safetensors",
|
| 157 |
+
"transformer.layers.21.self_attn.q_proj.weight": "model-00041-of-00094.safetensors",
|
| 158 |
+
"transformer.layers.21.self_attn.v_proj.weight": "model-00041-of-00094.safetensors",
|
| 159 |
+
"transformer.layers.22.input_layernorm.weight": "model-00041-of-00094.safetensors",
|
| 160 |
+
"transformer.layers.22.mlp.down_proj.weight": "model-00042-of-00094.safetensors",
|
| 161 |
+
"transformer.layers.22.mlp.gate_proj.weight": "model-00042-of-00094.safetensors",
|
| 162 |
+
"transformer.layers.22.mlp.up_proj.weight": "model-00043-of-00094.safetensors",
|
| 163 |
+
"transformer.layers.22.post_attention_layernorm.weight": "model-00043-of-00094.safetensors",
|
| 164 |
+
"transformer.layers.22.self_attn.k_proj.weight": "model-00043-of-00094.safetensors",
|
| 165 |
+
"transformer.layers.22.self_attn.o_proj.weight": "model-00043-of-00094.safetensors",
|
| 166 |
+
"transformer.layers.22.self_attn.q_proj.weight": "model-00043-of-00094.safetensors",
|
| 167 |
+
"transformer.layers.22.self_attn.v_proj.weight": "model-00043-of-00094.safetensors",
|
| 168 |
+
"transformer.layers.23.input_layernorm.weight": "model-00043-of-00094.safetensors",
|
| 169 |
+
"transformer.layers.23.mlp.down_proj.weight": "model-00044-of-00094.safetensors",
|
| 170 |
+
"transformer.layers.23.mlp.gate_proj.weight": "model-00044-of-00094.safetensors",
|
| 171 |
+
"transformer.layers.23.mlp.up_proj.weight": "model-00045-of-00094.safetensors",
|
| 172 |
+
"transformer.layers.23.post_attention_layernorm.weight": "model-00045-of-00094.safetensors",
|
| 173 |
+
"transformer.layers.23.self_attn.k_proj.weight": "model-00045-of-00094.safetensors",
|
| 174 |
+
"transformer.layers.23.self_attn.o_proj.weight": "model-00045-of-00094.safetensors",
|
| 175 |
+
"transformer.layers.23.self_attn.q_proj.weight": "model-00045-of-00094.safetensors",
|
| 176 |
+
"transformer.layers.23.self_attn.v_proj.weight": "model-00045-of-00094.safetensors",
|
| 177 |
+
"transformer.layers.24.input_layernorm.weight": "model-00045-of-00094.safetensors",
|
| 178 |
+
"transformer.layers.24.mlp.down_proj.weight": "model-00046-of-00094.safetensors",
|
| 179 |
+
"transformer.layers.24.mlp.gate_proj.weight": "model-00046-of-00094.safetensors",
|
| 180 |
+
"transformer.layers.24.mlp.up_proj.weight": "model-00047-of-00094.safetensors",
|
| 181 |
+
"transformer.layers.24.post_attention_layernorm.weight": "model-00047-of-00094.safetensors",
|
| 182 |
+
"transformer.layers.24.self_attn.k_proj.weight": "model-00047-of-00094.safetensors",
|
| 183 |
+
"transformer.layers.24.self_attn.o_proj.weight": "model-00047-of-00094.safetensors",
|
| 184 |
+
"transformer.layers.24.self_attn.q_proj.weight": "model-00047-of-00094.safetensors",
|
| 185 |
+
"transformer.layers.24.self_attn.v_proj.weight": "model-00047-of-00094.safetensors",
|
| 186 |
+
"transformer.layers.25.input_layernorm.weight": "model-00047-of-00094.safetensors",
|
| 187 |
+
"transformer.layers.25.mlp.down_proj.weight": "model-00048-of-00094.safetensors",
|
| 188 |
+
"transformer.layers.25.mlp.gate_proj.weight": "model-00048-of-00094.safetensors",
|
| 189 |
+
"transformer.layers.25.mlp.up_proj.weight": "model-00049-of-00094.safetensors",
|
| 190 |
+
"transformer.layers.25.post_attention_layernorm.weight": "model-00049-of-00094.safetensors",
|
| 191 |
+
"transformer.layers.25.self_attn.k_proj.weight": "model-00049-of-00094.safetensors",
|
| 192 |
+
"transformer.layers.25.self_attn.o_proj.weight": "model-00049-of-00094.safetensors",
|
| 193 |
+
"transformer.layers.25.self_attn.q_proj.weight": "model-00049-of-00094.safetensors",
|
| 194 |
+
"transformer.layers.25.self_attn.v_proj.weight": "model-00049-of-00094.safetensors",
|
| 195 |
+
"transformer.layers.26.input_layernorm.weight": "model-00049-of-00094.safetensors",
|
| 196 |
+
"transformer.layers.26.mlp.down_proj.weight": "model-00050-of-00094.safetensors",
|
| 197 |
+
"transformer.layers.26.mlp.gate_proj.weight": "model-00050-of-00094.safetensors",
|
| 198 |
+
"transformer.layers.26.mlp.up_proj.weight": "model-00051-of-00094.safetensors",
|
| 199 |
+
"transformer.layers.26.post_attention_layernorm.weight": "model-00051-of-00094.safetensors",
|
| 200 |
+
"transformer.layers.26.self_attn.k_proj.weight": "model-00051-of-00094.safetensors",
|
| 201 |
+
"transformer.layers.26.self_attn.o_proj.weight": "model-00051-of-00094.safetensors",
|
| 202 |
+
"transformer.layers.26.self_attn.q_proj.weight": "model-00051-of-00094.safetensors",
|
| 203 |
+
"transformer.layers.26.self_attn.v_proj.weight": "model-00051-of-00094.safetensors",
|
| 204 |
+
"transformer.layers.27.input_layernorm.weight": "model-00051-of-00094.safetensors",
|
| 205 |
+
"transformer.layers.27.mlp.down_proj.weight": "model-00052-of-00094.safetensors",
|
| 206 |
+
"transformer.layers.27.mlp.gate_proj.weight": "model-00052-of-00094.safetensors",
|
| 207 |
+
"transformer.layers.27.mlp.up_proj.weight": "model-00053-of-00094.safetensors",
|
| 208 |
+
"transformer.layers.27.post_attention_layernorm.weight": "model-00053-of-00094.safetensors",
|
| 209 |
+
"transformer.layers.27.self_attn.k_proj.weight": "model-00053-of-00094.safetensors",
|
| 210 |
+
"transformer.layers.27.self_attn.o_proj.weight": "model-00053-of-00094.safetensors",
|
| 211 |
+
"transformer.layers.27.self_attn.q_proj.weight": "model-00053-of-00094.safetensors",
|
| 212 |
+
"transformer.layers.27.self_attn.v_proj.weight": "model-00053-of-00094.safetensors",
|
| 213 |
+
"transformer.layers.3.input_layernorm.weight": "model-00053-of-00094.safetensors",
|
| 214 |
+
"transformer.layers.3.mlp.down_proj.weight": "model-00054-of-00094.safetensors",
|
| 215 |
+
"transformer.layers.3.mlp.gate_proj.weight": "model-00054-of-00094.safetensors",
|
| 216 |
+
"transformer.layers.3.mlp.up_proj.weight": "model-00055-of-00094.safetensors",
|
| 217 |
+
"transformer.layers.3.post_attention_layernorm.weight": "model-00055-of-00094.safetensors",
|
| 218 |
+
"transformer.layers.3.self_attn.k_proj.weight": "model-00055-of-00094.safetensors",
|
| 219 |
+
"transformer.layers.3.self_attn.o_proj.weight": "model-00055-of-00094.safetensors",
|
| 220 |
+
"transformer.layers.3.self_attn.q_proj.weight": "model-00055-of-00094.safetensors",
|
| 221 |
+
"transformer.layers.3.self_attn.v_proj.weight": "model-00055-of-00094.safetensors",
|
| 222 |
+
"transformer.layers.4.input_layernorm.weight": "model-00055-of-00094.safetensors",
|
| 223 |
+
"transformer.layers.4.mlp.down_proj.weight": "model-00056-of-00094.safetensors",
|
| 224 |
+
"transformer.layers.4.mlp.gate_proj.weight": "model-00056-of-00094.safetensors",
|
| 225 |
+
"transformer.layers.4.mlp.up_proj.weight": "model-00057-of-00094.safetensors",
|
| 226 |
+
"transformer.layers.4.post_attention_layernorm.weight": "model-00057-of-00094.safetensors",
|
| 227 |
+
"transformer.layers.4.self_attn.k_proj.weight": "model-00057-of-00094.safetensors",
|
| 228 |
+
"transformer.layers.4.self_attn.o_proj.weight": "model-00057-of-00094.safetensors",
|
| 229 |
+
"transformer.layers.4.self_attn.q_proj.weight": "model-00057-of-00094.safetensors",
|
| 230 |
+
"transformer.layers.4.self_attn.v_proj.weight": "model-00057-of-00094.safetensors",
|
| 231 |
+
"transformer.layers.5.input_layernorm.weight": "model-00057-of-00094.safetensors",
|
| 232 |
+
"transformer.layers.5.mlp.down_proj.weight": "model-00058-of-00094.safetensors",
|
| 233 |
+
"transformer.layers.5.mlp.gate_proj.weight": "model-00058-of-00094.safetensors",
|
| 234 |
+
"transformer.layers.5.mlp.up_proj.weight": "model-00059-of-00094.safetensors",
|
| 235 |
+
"transformer.layers.5.post_attention_layernorm.weight": "model-00059-of-00094.safetensors",
|
| 236 |
+
"transformer.layers.5.self_attn.k_proj.weight": "model-00059-of-00094.safetensors",
|
| 237 |
+
"transformer.layers.5.self_attn.o_proj.weight": "model-00059-of-00094.safetensors",
|
| 238 |
+
"transformer.layers.5.self_attn.q_proj.weight": "model-00059-of-00094.safetensors",
|
| 239 |
+
"transformer.layers.5.self_attn.v_proj.weight": "model-00059-of-00094.safetensors",
|
| 240 |
+
"transformer.layers.6.input_layernorm.weight": "model-00059-of-00094.safetensors",
|
| 241 |
+
"transformer.layers.6.mlp.down_proj.weight": "model-00060-of-00094.safetensors",
|
| 242 |
+
"transformer.layers.6.mlp.gate_proj.weight": "model-00060-of-00094.safetensors",
|
| 243 |
+
"transformer.layers.6.mlp.up_proj.weight": "model-00061-of-00094.safetensors",
|
| 244 |
+
"transformer.layers.6.post_attention_layernorm.weight": "model-00061-of-00094.safetensors",
|
| 245 |
+
"transformer.layers.6.self_attn.k_proj.weight": "model-00061-of-00094.safetensors",
|
| 246 |
+
"transformer.layers.6.self_attn.o_proj.weight": "model-00061-of-00094.safetensors",
|
| 247 |
+
"transformer.layers.6.self_attn.q_proj.weight": "model-00061-of-00094.safetensors",
|
| 248 |
+
"transformer.layers.6.self_attn.v_proj.weight": "model-00061-of-00094.safetensors",
|
| 249 |
+
"transformer.layers.7.input_layernorm.weight": "model-00061-of-00094.safetensors",
|
| 250 |
+
"transformer.layers.7.mlp.down_proj.weight": "model-00062-of-00094.safetensors",
|
| 251 |
+
"transformer.layers.7.mlp.gate_proj.weight": "model-00062-of-00094.safetensors",
|
| 252 |
+
"transformer.layers.7.mlp.up_proj.weight": "model-00063-of-00094.safetensors",
|
| 253 |
+
"transformer.layers.7.post_attention_layernorm.weight": "model-00063-of-00094.safetensors",
|
| 254 |
+
"transformer.layers.7.self_attn.k_proj.weight": "model-00063-of-00094.safetensors",
|
| 255 |
+
"transformer.layers.7.self_attn.o_proj.weight": "model-00063-of-00094.safetensors",
|
| 256 |
+
"transformer.layers.7.self_attn.q_proj.weight": "model-00063-of-00094.safetensors",
|
| 257 |
+
"transformer.layers.7.self_attn.v_proj.weight": "model-00063-of-00094.safetensors",
|
| 258 |
+
"transformer.layers.8.input_layernorm.weight": "model-00063-of-00094.safetensors",
|
| 259 |
+
"transformer.layers.8.mlp.down_proj.weight": "model-00064-of-00094.safetensors",
|
| 260 |
+
"transformer.layers.8.mlp.gate_proj.weight": "model-00064-of-00094.safetensors",
|
| 261 |
+
"transformer.layers.8.mlp.up_proj.weight": "model-00065-of-00094.safetensors",
|
| 262 |
+
"transformer.layers.8.post_attention_layernorm.weight": "model-00065-of-00094.safetensors",
|
| 263 |
+
"transformer.layers.8.self_attn.k_proj.weight": "model-00065-of-00094.safetensors",
|
| 264 |
+
"transformer.layers.8.self_attn.o_proj.weight": "model-00065-of-00094.safetensors",
|
| 265 |
+
"transformer.layers.8.self_attn.q_proj.weight": "model-00065-of-00094.safetensors",
|
| 266 |
+
"transformer.layers.8.self_attn.v_proj.weight": "model-00065-of-00094.safetensors",
|
| 267 |
+
"transformer.layers.9.input_layernorm.weight": "model-00065-of-00094.safetensors",
|
| 268 |
+
"transformer.layers.9.mlp.down_proj.weight": "model-00066-of-00094.safetensors",
|
| 269 |
+
"transformer.layers.9.mlp.gate_proj.weight": "model-00066-of-00094.safetensors",
|
| 270 |
+
"transformer.layers.9.mlp.up_proj.weight": "model-00067-of-00094.safetensors",
|
| 271 |
+
"transformer.layers.9.post_attention_layernorm.weight": "model-00067-of-00094.safetensors",
|
| 272 |
+
"transformer.layers.9.self_attn.k_proj.weight": "model-00067-of-00094.safetensors",
|
| 273 |
+
"transformer.layers.9.self_attn.o_proj.weight": "model-00067-of-00094.safetensors",
|
| 274 |
+
"transformer.layers.9.self_attn.q_proj.weight": "model-00067-of-00094.safetensors",
|
| 275 |
+
"transformer.layers.9.self_attn.v_proj.weight": "model-00067-of-00094.safetensors",
|
| 276 |
+
"transformer.norm.weight": "model-00067-of-00094.safetensors",
|
| 277 |
+
"transformer2.embed_tokens.weight": "model-00068-of-00094.safetensors",
|
| 278 |
+
"transformer2.layers.0.input_layernorm.weight": "model-00068-of-00094.safetensors",
|
| 279 |
+
"transformer2.layers.0.mlp.down_proj.weight": "model-00069-of-00094.safetensors",
|
| 280 |
+
"transformer2.layers.0.mlp.gate_proj.weight": "model-00069-of-00094.safetensors",
|
| 281 |
+
"transformer2.layers.0.mlp.up_proj.weight": "model-00070-of-00094.safetensors",
|
| 282 |
+
"transformer2.layers.0.post_attention_layernorm.weight": "model-00070-of-00094.safetensors",
|
| 283 |
+
"transformer2.layers.0.self_attn.k_proj.weight": "model-00070-of-00094.safetensors",
|
| 284 |
+
"transformer2.layers.0.self_attn.o_proj.weight": "model-00070-of-00094.safetensors",
|
| 285 |
+
"transformer2.layers.0.self_attn.q_proj.weight": "model-00070-of-00094.safetensors",
|
| 286 |
+
"transformer2.layers.0.self_attn.v_proj.weight": "model-00070-of-00094.safetensors",
|
| 287 |
+
"transformer2.layers.1.input_layernorm.weight": "model-00070-of-00094.safetensors",
|
| 288 |
+
"transformer2.layers.1.mlp.down_proj.weight": "model-00071-of-00094.safetensors",
|
| 289 |
+
"transformer2.layers.1.mlp.gate_proj.weight": "model-00071-of-00094.safetensors",
|
| 290 |
+
"transformer2.layers.1.mlp.up_proj.weight": "model-00072-of-00094.safetensors",
|
| 291 |
+
"transformer2.layers.1.post_attention_layernorm.weight": "model-00072-of-00094.safetensors",
|
| 292 |
+
"transformer2.layers.1.self_attn.k_proj.weight": "model-00072-of-00094.safetensors",
|
| 293 |
+
"transformer2.layers.1.self_attn.o_proj.weight": "model-00072-of-00094.safetensors",
|
| 294 |
+
"transformer2.layers.1.self_attn.q_proj.weight": "model-00072-of-00094.safetensors",
|
| 295 |
+
"transformer2.layers.1.self_attn.v_proj.weight": "model-00072-of-00094.safetensors",
|
| 296 |
+
"transformer2.layers.10.input_layernorm.weight": "model-00072-of-00094.safetensors",
|
| 297 |
+
"transformer2.layers.10.mlp.down_proj.weight": "model-00073-of-00094.safetensors",
|
| 298 |
+
"transformer2.layers.10.mlp.gate_proj.weight": "model-00073-of-00094.safetensors",
|
| 299 |
+
"transformer2.layers.10.mlp.up_proj.weight": "model-00074-of-00094.safetensors",
|
| 300 |
+
"transformer2.layers.10.post_attention_layernorm.weight": "model-00074-of-00094.safetensors",
|
| 301 |
+
"transformer2.layers.10.self_attn.k_proj.weight": "model-00074-of-00094.safetensors",
|
| 302 |
+
"transformer2.layers.10.self_attn.o_proj.weight": "model-00074-of-00094.safetensors",
|
| 303 |
+
"transformer2.layers.10.self_attn.q_proj.weight": "model-00074-of-00094.safetensors",
|
| 304 |
+
"transformer2.layers.10.self_attn.v_proj.weight": "model-00074-of-00094.safetensors",
|
| 305 |
+
"transformer2.layers.11.input_layernorm.weight": "model-00074-of-00094.safetensors",
|
| 306 |
+
"transformer2.layers.11.mlp.down_proj.weight": "model-00075-of-00094.safetensors",
|
| 307 |
+
"transformer2.layers.11.mlp.gate_proj.weight": "model-00075-of-00094.safetensors",
|
| 308 |
+
"transformer2.layers.11.mlp.up_proj.weight": "model-00076-of-00094.safetensors",
|
| 309 |
+
"transformer2.layers.11.post_attention_layernorm.weight": "model-00076-of-00094.safetensors",
|
| 310 |
+
"transformer2.layers.11.self_attn.k_proj.weight": "model-00076-of-00094.safetensors",
|
| 311 |
+
"transformer2.layers.11.self_attn.o_proj.weight": "model-00076-of-00094.safetensors",
|
| 312 |
+
"transformer2.layers.11.self_attn.q_proj.weight": "model-00076-of-00094.safetensors",
|
| 313 |
+
"transformer2.layers.11.self_attn.v_proj.weight": "model-00076-of-00094.safetensors",
|
| 314 |
+
"transformer2.layers.2.input_layernorm.weight": "model-00076-of-00094.safetensors",
|
| 315 |
+
"transformer2.layers.2.mlp.down_proj.weight": "model-00077-of-00094.safetensors",
|
| 316 |
+
"transformer2.layers.2.mlp.gate_proj.weight": "model-00077-of-00094.safetensors",
|
| 317 |
+
"transformer2.layers.2.mlp.up_proj.weight": "model-00078-of-00094.safetensors",
|
| 318 |
+
"transformer2.layers.2.post_attention_layernorm.weight": "model-00078-of-00094.safetensors",
|
| 319 |
+
"transformer2.layers.2.self_attn.k_proj.weight": "model-00078-of-00094.safetensors",
|
| 320 |
+
"transformer2.layers.2.self_attn.o_proj.weight": "model-00078-of-00094.safetensors",
|
| 321 |
+
"transformer2.layers.2.self_attn.q_proj.weight": "model-00078-of-00094.safetensors",
|
| 322 |
+
"transformer2.layers.2.self_attn.v_proj.weight": "model-00078-of-00094.safetensors",
|
| 323 |
+
"transformer2.layers.3.input_layernorm.weight": "model-00078-of-00094.safetensors",
|
| 324 |
+
"transformer2.layers.3.mlp.down_proj.weight": "model-00079-of-00094.safetensors",
|
| 325 |
+
"transformer2.layers.3.mlp.gate_proj.weight": "model-00079-of-00094.safetensors",
|
| 326 |
+
"transformer2.layers.3.mlp.up_proj.weight": "model-00080-of-00094.safetensors",
|
| 327 |
+
"transformer2.layers.3.post_attention_layernorm.weight": "model-00080-of-00094.safetensors",
|
| 328 |
+
"transformer2.layers.3.self_attn.k_proj.weight": "model-00080-of-00094.safetensors",
|
| 329 |
+
"transformer2.layers.3.self_attn.o_proj.weight": "model-00080-of-00094.safetensors",
|
| 330 |
+
"transformer2.layers.3.self_attn.q_proj.weight": "model-00080-of-00094.safetensors",
|
| 331 |
+
"transformer2.layers.3.self_attn.v_proj.weight": "model-00080-of-00094.safetensors",
|
| 332 |
+
"transformer2.layers.4.input_layernorm.weight": "model-00080-of-00094.safetensors",
|
| 333 |
+
"transformer2.layers.4.mlp.down_proj.weight": "model-00081-of-00094.safetensors",
|
| 334 |
+
"transformer2.layers.4.mlp.gate_proj.weight": "model-00081-of-00094.safetensors",
|
| 335 |
+
"transformer2.layers.4.mlp.up_proj.weight": "model-00082-of-00094.safetensors",
|
| 336 |
+
"transformer2.layers.4.post_attention_layernorm.weight": "model-00082-of-00094.safetensors",
|
| 337 |
+
"transformer2.layers.4.self_attn.k_proj.weight": "model-00082-of-00094.safetensors",
|
| 338 |
+
"transformer2.layers.4.self_attn.o_proj.weight": "model-00082-of-00094.safetensors",
|
| 339 |
+
"transformer2.layers.4.self_attn.q_proj.weight": "model-00082-of-00094.safetensors",
|
| 340 |
+
"transformer2.layers.4.self_attn.v_proj.weight": "model-00082-of-00094.safetensors",
|
| 341 |
+
"transformer2.layers.5.input_layernorm.weight": "model-00082-of-00094.safetensors",
|
| 342 |
+
"transformer2.layers.5.mlp.down_proj.weight": "model-00083-of-00094.safetensors",
|
| 343 |
+
"transformer2.layers.5.mlp.gate_proj.weight": "model-00083-of-00094.safetensors",
|
| 344 |
+
"transformer2.layers.5.mlp.up_proj.weight": "model-00084-of-00094.safetensors",
|
| 345 |
+
"transformer2.layers.5.post_attention_layernorm.weight": "model-00084-of-00094.safetensors",
|
| 346 |
+
"transformer2.layers.5.self_attn.k_proj.weight": "model-00084-of-00094.safetensors",
|
| 347 |
+
"transformer2.layers.5.self_attn.o_proj.weight": "model-00084-of-00094.safetensors",
|
| 348 |
+
"transformer2.layers.5.self_attn.q_proj.weight": "model-00084-of-00094.safetensors",
|
| 349 |
+
"transformer2.layers.5.self_attn.v_proj.weight": "model-00084-of-00094.safetensors",
|
| 350 |
+
"transformer2.layers.6.input_layernorm.weight": "model-00084-of-00094.safetensors",
|
| 351 |
+
"transformer2.layers.6.mlp.down_proj.weight": "model-00085-of-00094.safetensors",
|
| 352 |
+
"transformer2.layers.6.mlp.gate_proj.weight": "model-00085-of-00094.safetensors",
|
| 353 |
+
"transformer2.layers.6.mlp.up_proj.weight": "model-00086-of-00094.safetensors",
|
| 354 |
+
"transformer2.layers.6.post_attention_layernorm.weight": "model-00086-of-00094.safetensors",
|
| 355 |
+
"transformer2.layers.6.self_attn.k_proj.weight": "model-00086-of-00094.safetensors",
|
| 356 |
+
"transformer2.layers.6.self_attn.o_proj.weight": "model-00086-of-00094.safetensors",
|
| 357 |
+
"transformer2.layers.6.self_attn.q_proj.weight": "model-00086-of-00094.safetensors",
|
| 358 |
+
"transformer2.layers.6.self_attn.v_proj.weight": "model-00086-of-00094.safetensors",
|
| 359 |
+
"transformer2.layers.7.input_layernorm.weight": "model-00086-of-00094.safetensors",
|
| 360 |
+
"transformer2.layers.7.mlp.down_proj.weight": "model-00087-of-00094.safetensors",
|
| 361 |
+
"transformer2.layers.7.mlp.gate_proj.weight": "model-00087-of-00094.safetensors",
|
| 362 |
+
"transformer2.layers.7.mlp.up_proj.weight": "model-00088-of-00094.safetensors",
|
| 363 |
+
"transformer2.layers.7.post_attention_layernorm.weight": "model-00088-of-00094.safetensors",
|
| 364 |
+
"transformer2.layers.7.self_attn.k_proj.weight": "model-00088-of-00094.safetensors",
|
| 365 |
+
"transformer2.layers.7.self_attn.o_proj.weight": "model-00088-of-00094.safetensors",
|
| 366 |
+
"transformer2.layers.7.self_attn.q_proj.weight": "model-00088-of-00094.safetensors",
|
| 367 |
+
"transformer2.layers.7.self_attn.v_proj.weight": "model-00088-of-00094.safetensors",
|
| 368 |
+
"transformer2.layers.8.input_layernorm.weight": "model-00088-of-00094.safetensors",
|
| 369 |
+
"transformer2.layers.8.mlp.down_proj.weight": "model-00089-of-00094.safetensors",
|
| 370 |
+
"transformer2.layers.8.mlp.gate_proj.weight": "model-00089-of-00094.safetensors",
|
| 371 |
+
"transformer2.layers.8.mlp.up_proj.weight": "model-00090-of-00094.safetensors",
|
| 372 |
+
"transformer2.layers.8.post_attention_layernorm.weight": "model-00090-of-00094.safetensors",
|
| 373 |
+
"transformer2.layers.8.self_attn.k_proj.weight": "model-00090-of-00094.safetensors",
|
| 374 |
+
"transformer2.layers.8.self_attn.o_proj.weight": "model-00090-of-00094.safetensors",
|
| 375 |
+
"transformer2.layers.8.self_attn.q_proj.weight": "model-00090-of-00094.safetensors",
|
| 376 |
+
"transformer2.layers.8.self_attn.v_proj.weight": "model-00090-of-00094.safetensors",
|
| 377 |
+
"transformer2.layers.9.input_layernorm.weight": "model-00090-of-00094.safetensors",
|
| 378 |
+
"transformer2.layers.9.mlp.down_proj.weight": "model-00091-of-00094.safetensors",
|
| 379 |
+
"transformer2.layers.9.mlp.gate_proj.weight": "model-00091-of-00094.safetensors",
|
| 380 |
+
"transformer2.layers.9.mlp.up_proj.weight": "model-00092-of-00094.safetensors",
|
| 381 |
+
"transformer2.layers.9.post_attention_layernorm.weight": "model-00092-of-00094.safetensors",
|
| 382 |
+
"transformer2.layers.9.self_attn.k_proj.weight": "model-00092-of-00094.safetensors",
|
| 383 |
+
"transformer2.layers.9.self_attn.o_proj.weight": "model-00092-of-00094.safetensors",
|
| 384 |
+
"transformer2.layers.9.self_attn.q_proj.weight": "model-00092-of-00094.safetensors",
|
| 385 |
+
"transformer2.layers.9.self_attn.v_proj.weight": "model-00092-of-00094.safetensors",
|
| 386 |
+
"transformer2.norm.weight": "model-00092-of-00094.safetensors",
|
| 387 |
+
"transformer_lm_head.weight": "model-00093-of-00094.safetensors",
|
| 388 |
+
"type_info_output_proj.weight": "model-00094-of-00094.safetensors"
|
| 389 |
+
}
|
| 390 |
+
}
|
vocab.yaml
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- '[verse]'
|
| 2 |
+
- '[chorus]'
|
| 3 |
+
- '[bridge]'
|
| 4 |
+
- '[intro-short]'
|
| 5 |
+
- '[intro-medium]'
|
| 6 |
+
- '[intro-long]'
|
| 7 |
+
- '[outro-short]'
|
| 8 |
+
- '[outro-medium]'
|
| 9 |
+
- '[outro-long]'
|
| 10 |
+
- '[inst-short]'
|
| 11 |
+
- '[inst-medium]'
|
| 12 |
+
- '[inst-long]'
|
| 13 |
+
- '[silence]'
|