OriDragon2000 commited on
Commit
e7a66e6
1 Parent(s): d174176

Upload LlamaForCausalLM

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -20,7 +20,7 @@
20
  "quantization_config": {
21
  "_load_in_4bit": true,
22
  "_load_in_8bit": false,
23
- "bnb_4bit_compute_dtype": "float32",
24
  "bnb_4bit_quant_storage": "uint8",
25
  "bnb_4bit_quant_type": "nf4",
26
  "bnb_4bit_use_double_quant": false,
 
20
  "quantization_config": {
21
  "_load_in_4bit": true,
22
  "_load_in_8bit": false,
23
+ "bnb_4bit_compute_dtype": "bfloat16",
24
  "bnb_4bit_quant_storage": "uint8",
25
  "bnb_4bit_quant_type": "nf4",
26
  "bnb_4bit_use_double_quant": false,