ZahrizhalAli commited on
Commit
ade2efc
1 Parent(s): 8899939

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +15 -2
config.json CHANGED
@@ -1,4 +1,4 @@
1
- {
2
  "_name_or_path": "ZahrizhalAli/llama-7b-code-generation",
3
  "architectures": [
4
  "LlamaForCausalLM"
@@ -15,11 +15,24 @@
15
  "num_hidden_layers": 32,
16
  "num_key_value_heads": 32,
17
  "pretraining_tp": 1,
 
 
 
 
 
 
 
 
 
 
 
 
18
  "rms_norm_eps": 1e-05,
19
  "rope_scaling": null,
 
20
  "tie_word_embeddings": false,
21
  "torch_dtype": "float16",
22
- "transformers_version": "4.31.0.dev0",
23
  "use_cache": true,
24
  "vocab_size": 32000
25
  }
 
1
+ LlamaConfig {
2
  "_name_or_path": "ZahrizhalAli/llama-7b-code-generation",
3
  "architectures": [
4
  "LlamaForCausalLM"
 
15
  "num_hidden_layers": 32,
16
  "num_key_value_heads": 32,
17
  "pretraining_tp": 1,
18
+ "quantization_config": {
19
+ "bnb_4bit_compute_dtype": "bfloat16",
20
+ "bnb_4bit_quant_type": "nf4",
21
+ "bnb_4bit_use_double_quant": true,
22
+ "llm_int8_enable_fp32_cpu_offload": false,
23
+ "llm_int8_has_fp16_weight": false,
24
+ "llm_int8_skip_modules": null,
25
+ "llm_int8_threshold": 6.0,
26
+ "load_in_4bit": true,
27
+ "load_in_8bit": false,
28
+ "quant_method": "bitsandbytes"
29
+ },
30
  "rms_norm_eps": 1e-05,
31
  "rope_scaling": null,
32
+ "rope_theta": 10000.0,
33
  "tie_word_embeddings": false,
34
  "torch_dtype": "float16",
35
+ "transformers_version": "4.33.2",
36
  "use_cache": true,
37
  "vocab_size": 32000
38
  }