bavihao commited on
Commit
a875700
·
verified ·
1 Parent(s): ac048cd

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +15 -0
config.json CHANGED
@@ -42,6 +42,21 @@
42
  "num_hidden_layers": 32,
43
  "num_mel_bins": 80,
44
  "pad_token_id": 50257,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
  "scale_embedding": false,
46
  "torch_dtype": "float16",
47
  "transformers_version": "4.42.0.dev0",
 
42
  "num_hidden_layers": 32,
43
  "num_mel_bins": 80,
44
  "pad_token_id": 50257,
45
+ "quantization_config": {
46
+ "_load_in_4bit": false,
47
+ "_load_in_8bit": true,
48
+ "bnb_4bit_compute_dtype": "float32",
49
+ "bnb_4bit_quant_storage": "uint8",
50
+ "bnb_4bit_quant_type": "fp4",
51
+ "bnb_4bit_use_double_quant": false,
52
+ "llm_int8_enable_fp32_cpu_offload": false,
53
+ "llm_int8_has_fp16_weight": false,
54
+ "llm_int8_skip_modules": null,
55
+ "llm_int8_threshold": 6.0,
56
+ "load_in_4bit": false,
57
+ "load_in_8bit": true,
58
+ "quant_method": "bitsandbytes"
59
+ },
60
  "scale_embedding": false,
61
  "torch_dtype": "float16",
62
  "transformers_version": "4.42.0.dev0",