mostafatarek4 commited on
Commit
ede5a88
·
verified ·
1 Parent(s): bfcce4f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +24 -24
config.json CHANGED
@@ -1,25 +1,25 @@
1
  {
2
- "model_type": "llama",
3
- "hidden_size": 4096, // Adjust based on LLaMA size (8B uses 4096)
4
- "intermediate_size": 11008, // Typically 2.67x the hidden size
5
- "num_attention_heads": 32, // Adjust based on model
6
- "num_hidden_layers": 32, // Number of transformer layers
7
- "vocab_size": 32000, // Common LLaMA vocab size
8
- "max_position_embeddings": 2048, // Max sequence length you used
9
- "rope_scaling": {
10
- "type": "dynamic",
11
- "factor": 2.0
12
- },
13
- "rotary_embedding_base": 10000, // Used for RoPE embeddings
14
- "tie_word_embeddings": false, // Generally false for LLaMA
15
- "torch_dtype": "float16", // Use float16 for efficient inference
16
- "use_cache": true, // Enable cache for inference
17
- "pad_token_id": 0,
18
- "bos_token_id": 1,
19
- "eos_token_id": 2,
20
- "quantization_config": {
21
- "load_in_4bit": true,
22
- "bnb_4bit_use_double_quant": true,
23
- "bnb_4bit_quant_type": "nf4"
24
- }
25
- }
 
1
  {
2
+ "model_type": "llama",
3
+ "hidden_size": 4096,
4
+ "intermediate_size": 11008,
5
+ "num_attention_heads": 32,
6
+ "num_hidden_layers": 32,
7
+ "vocab_size": 32000,
8
+ "max_position_embeddings": 2048,
9
+ "rope_scaling": {
10
+ "type": "dynamic",
11
+ "factor": 2
12
+ },
13
+ "rotary_embedding_base": 10000,
14
+ "tie_word_embeddings": false,
15
+ "torch_dtype": "float16",
16
+ "use_cache": true,
17
+ "pad_token_id": 0,
18
+ "bos_token_id": 1,
19
+ "eos_token_id": 2,
20
+ "quantization_config": {
21
+ "load_in_4bit": true,
22
+ "bnb_4bit_use_double_quant": true,
23
+ "bnb_4bit_quant_type": "nf4"
24
+ }
25
+ }