shreyask commited on
Commit
c167f01
·
verified ·
1 Parent(s): c524ad2

Set kv_cache_dtype + torch_dtype to float32 (matches actual ONNX weights)

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -16,7 +16,7 @@
16
  "conv_bias": false,
17
  "conv_dim": 2048,
18
  "conv_use_xavier_init": true,
19
- "torch_dtype": "float16",
20
  "eos_token_id": 7,
21
  "full_attn_idxs": null,
22
  "hidden_size": 2048,
@@ -63,7 +63,7 @@
63
  "model_q4.onnx": 1
64
  },
65
  "kv_cache_dtype": {
66
- "q4": "float16"
67
  }
68
  }
69
  }
 
16
  "conv_bias": false,
17
  "conv_dim": 2048,
18
  "conv_use_xavier_init": true,
19
+ "torch_dtype": "float32",
20
  "eos_token_id": 7,
21
  "full_attn_idxs": null,
22
  "hidden_size": 2048,
 
63
  "model_q4.onnx": 1
64
  },
65
  "kv_cache_dtype": {
66
+ "q4": "float32"
67
  }
68
  }
69
  }