diff options
Diffstat (limited to 'configs/model')
| -rw-r--r-- | configs/model/l350.yaml | 7 | ||||
| -rw-r--r-- | configs/model/m124.yaml | 7 | ||||
| -rw-r--r-- | configs/model/s60.yaml | 7 |
3 files changed, 21 insertions, 0 deletions
diff --git a/configs/model/l350.yaml b/configs/model/l350.yaml new file mode 100644 index 0000000..cd36f07 --- /dev/null +++ b/configs/model/l350.yaml @@ -0,0 +1,7 @@ +d: 1024 +layers: 24 +heads: 16 +vocab: 50257 +seq_len: 1024 +ffn_mult: 8/3 +pos: learned diff --git a/configs/model/m124.yaml b/configs/model/m124.yaml new file mode 100644 index 0000000..ad3b5fa --- /dev/null +++ b/configs/model/m124.yaml @@ -0,0 +1,7 @@ +d: 768 +layers: 12 +heads: 12 +vocab: 50257 +seq_len: 1024 +ffn_mult: 8/3 +pos: learned diff --git a/configs/model/s60.yaml b/configs/model/s60.yaml new file mode 100644 index 0000000..ebccc0a --- /dev/null +++ b/configs/model/s60.yaml @@ -0,0 +1,7 @@ +d: 512 +layers: 8 +heads: 8 +vocab: 50257 +seq_len: 1024 +ffn_mult: 8/3 # SwiGLU: hidden = mult*d rounded to 64 +pos: learned |