mjschock commited on
Commit
3d03f45
1 Parent(s): 50d41cf

Upload model

Browse files
Files changed (2) hide show
  1. config.json +3 -2
  2. generation_config.json +4 -0
config.json CHANGED
@@ -1,10 +1,11 @@
1
  {
2
  "architectures": [
3
- "MambaModel"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_mamba.MambaConfig",
7
- "AutoModel": "modeling_mamba.MambaModel"
 
8
  },
9
  "d_model": 2048,
10
  "fused_add_norm": true,
 
1
  {
2
  "architectures": [
3
+ "MambaModelForCausalLM"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_mamba.MambaConfig",
7
+ "AutoModel": "modeling_mamba.MambaModel",
8
+ "AutoModelForCausalLM": "modeling_mamba.MambaModelForCausalLM"
9
  },
10
  "d_model": 2048,
11
  "fused_add_norm": true,
generation_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "transformers_version": "4.37.2"
4
+ }