Iker commited on
Commit
d3e41fe
1 Parent(s): c420a3f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -1
config.json CHANGED
@@ -3,6 +3,9 @@
3
  "architectures": [
4
  "LlamaForCausalLM"
5
  ],
 
 
 
6
  "attention_bias": false,
7
  "bos_token_id": 1,
8
  "eos_token_id": 2,
@@ -21,7 +24,7 @@
21
  "rope_theta": 1000000,
22
  "tie_word_embeddings": false,
23
  "torch_dtype": "bfloat16",
24
- "transformers_version": "4.34.0.dev0",
25
  "use_cache": true,
26
  "vocab_size": 32016
27
  }
 
3
  "architectures": [
4
  "LlamaForCausalLM"
5
  ],
6
+ "auto_map": {
7
+ "AutoModelForCausalLM": "modeling_flash_llama.LlamaForCausalLM"
8
+ },
9
  "attention_bias": false,
10
  "bos_token_id": 1,
11
  "eos_token_id": 2,
 
24
  "rope_theta": 1000000,
25
  "tie_word_embeddings": false,
26
  "torch_dtype": "bfloat16",
27
+ "transformers_version": "4.33.1",
28
  "use_cache": true,
29
  "vocab_size": 32016
30
  }