Upload folder using huggingface_hub
Browse files- README.md +8 -0
- config.json +11 -0
- model.safetensors +3 -0
- modeling_selection.py +1 -0
- pytorch_model.bin +3 -0
- tokenizer.json +0 -0
- tokenizer_config.json +13 -0
README.md
CHANGED
|
@@ -1,3 +1,11 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- code
|
| 6 |
+
- selection-mixing
|
| 7 |
---
|
| 8 |
+
|
| 9 |
+
# selection-coder
|
| 10 |
+
|
| 11 |
+
~211M decoder, **only SelectionMixing (order-statistic routing), no attention**. Muon, bf16, trained on codeparrot/codeparrot-clean. step=2000.
|
config.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"vocab": 32768,
|
| 3 |
+
"d": 1024,
|
| 4 |
+
"layers": 12,
|
| 5 |
+
"heads": 16,
|
| 6 |
+
"seq": 512,
|
| 7 |
+
"ff": 4,
|
| 8 |
+
"arch": "SelectionCoder",
|
| 9 |
+
"mixer": "SelectionMixing(no-attention)",
|
| 10 |
+
"step": 2000
|
| 11 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5093c7e29bf090517fd52b456afa6fc156e853d29a6fe7fda3dd316a3aba40b7
|
| 3 |
+
size 488715288
|
modeling_selection.py
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
# architecture defined in train_coder.py (SelectionCoder / SelectionMixing)
|
pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f6ed5264e82ea91e3bc38d77b4f5065e830e80484563d6efafab63512eb3e084
|
| 3 |
+
size 489804083
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"backend": "tokenizers",
|
| 4 |
+
"bos_token": "<|endoftext|>",
|
| 5 |
+
"eos_token": "<|endoftext|>",
|
| 6 |
+
"errors": "replace",
|
| 7 |
+
"is_local": false,
|
| 8 |
+
"local_files_only": false,
|
| 9 |
+
"model_max_length": 1024,
|
| 10 |
+
"pad_token": null,
|
| 11 |
+
"tokenizer_class": "GPT2Tokenizer",
|
| 12 |
+
"unk_token": "<|endoftext|>"
|
| 13 |
+
}
|