ericrisco commited on
Commit
7ad0c7c
1 Parent(s): f7f285e

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -22,7 +22,7 @@
22
  "pad_token_id": 3,
23
  "pretraining_tp": 1,
24
  "quantization_config": {
25
- "bnb_4bit_compute_dtype": "float32",
26
  "bnb_4bit_quant_type": "nf4",
27
  "bnb_4bit_use_double_quant": false,
28
  "llm_int8_enable_fp32_cpu_offload": false,
@@ -37,7 +37,7 @@
37
  "skip_bias_add": true,
38
  "skip_bias_add_qkv": false,
39
  "slow_but_exact": false,
40
- "torch_dtype": "float32",
41
  "transformers_version": "4.37.2",
42
  "unk_token_id": 0,
43
  "use_cache": false,
 
22
  "pad_token_id": 3,
23
  "pretraining_tp": 1,
24
  "quantization_config": {
25
+ "bnb_4bit_compute_dtype": "float16",
26
  "bnb_4bit_quant_type": "nf4",
27
  "bnb_4bit_use_double_quant": false,
28
  "llm_int8_enable_fp32_cpu_offload": false,
 
37
  "skip_bias_add": true,
38
  "skip_bias_add_qkv": false,
39
  "slow_but_exact": false,
40
+ "torch_dtype": "float16",
41
  "transformers_version": "4.37.2",
42
  "unk_token_id": 0,
43
  "use_cache": false,