Upload folder using huggingface_hub
Browse files- README.md +5 -0
- ae_final.pt +3 -0
- calm_final.pt +3 -0
- config.json +12 -0
README.md
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# CALM (Continuous Autoregressive Language Model)
|
| 2 |
+
Trained by Mohamed Yasser
|
| 3 |
+
|
| 4 |
+
This repository contains both Autoencoder (AE) and CALM latent predictor.
|
| 5 |
+
Inference code included below.
|
ae_final.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7dc14354166190807995240c8e3a276c5f3b5f731a012272155db884de666df9
|
| 3 |
+
size 112118872
|
calm_final.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a7f8be803ed313ccf14c44b22bf2a128584f02564a9a85110a68c0d778401789
|
| 3 |
+
size 220895745
|
config.json
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "CALM",
|
| 3 |
+
"ae_hidden_dim": 512,
|
| 4 |
+
"ae_latent_dim": 128,
|
| 5 |
+
"ae_chunk_size_k": 4,
|
| 6 |
+
"lm_hidden_dim": 512,
|
| 7 |
+
"lm_ffn_dim": 2048,
|
| 8 |
+
"lm_num_layers": 8,
|
| 9 |
+
"lm_num_heads": 8,
|
| 10 |
+
"vocab_size": 50257,
|
| 11 |
+
"framework": "PyTorch"
|
| 12 |
+
}
|