ericrisco commited on
Commit
be8ba92
1 Parent(s): 7df81b0

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -0
config.json CHANGED
@@ -22,6 +22,18 @@
22
  "pad_token_id": 3,
23
  "pretraining_tp": 1,
24
  "seq_length": 4096,
 
 
 
 
 
 
 
 
 
 
 
 
25
  "skip_bias_add": true,
26
  "skip_bias_add_qkv": false,
27
  "slow_but_exact": false,
 
22
  "pad_token_id": 3,
23
  "pretraining_tp": 1,
24
  "seq_length": 4096,
25
+ "quantization_config": {
26
+ "bnb_4bit_compute_dtype": "float16",
27
+ "bnb_4bit_quant_type": "nf4",
28
+ "bnb_4bit_use_double_quant": false,
29
+ "llm_int8_enable_fp32_cpu_offload": false,
30
+ "llm_int8_has_fp16_weight": false,
31
+ "llm_int8_skip_modules": null,
32
+ "llm_int8_threshold": 6.0,
33
+ "load_in_4bit": true,
34
+ "load_in_8bit": false,
35
+ "quant_method": "bitsandbytes"
36
+ },
37
  "skip_bias_add": true,
38
  "skip_bias_add_qkv": false,
39
  "slow_but_exact": false,