PatrickHaller commited on
Commit
34d4735
1 Parent(s): 6572850

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +35 -0
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "torch_dtype": "float32",
3
+ "architectures": [
4
+ "xLSTMForCausalLM"
5
+ ],
6
+ "transformers_version": "4.44.0",
7
+ "_xlstm_config": {
8
+ "num_blocks": 2,
9
+ "embedding_dim": 64,
10
+ "mlstm_block": {
11
+ "mlstm": {
12
+ "num_heads": 1
13
+ }
14
+ },
15
+ "slstm_block": {
16
+ "slstm": {
17
+ "num_heads": 1
18
+ }
19
+ },
20
+ "slstm_at": [
21
+ 1
22
+ ],
23
+ "context_length": 256,
24
+ "vocab_size": 32000
25
+ },
26
+ "vocab_size": 32000,
27
+ "embedding_dim": 64,
28
+ "context_length": 256,
29
+ "model_type": "xlstm",
30
+ "auto_map": {
31
+ "AutoConfig": "configuration_xlstm.xLSTMConfig",
32
+ "AutoModelForCausalLM": "modeling_xlstm.xLSTMForCausalLM",
33
+ "AutoModel": "modeling_xlstm.xLSTMModel"
34
+ }
35
+ }