Hugging Face's logo

Steveeeeeeen
/
whisper-large-v1-trt-llm-checkpoint

Model card Files Community
Steveeeeeeen HF staff commited on
Commit
90ee302
·
1 Parent(s): 7cbe9bb

Change back weight to float16

Browse files
Files changed (2) hide show
  1. decoder/config.json +2 -2
  2. encoder/config.json +1 -1
decoder/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "architecture": "DecoderModel",
3
- "dtype": "float32",
4
- "logits_dtype": "float32",
5
  "num_hidden_layers": 32,
6
  "num_attention_heads": 20,
7
  "hidden_size": 1280,
 
1
  {
2
  "architecture": "DecoderModel",
3
+ "dtype": "float16",
4
+ "logits_dtype": "float16",
5
  "num_hidden_layers": 32,
6
  "num_attention_heads": 20,
7
  "hidden_size": 1280,
encoder/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architecture": "WhisperEncoder",
3
- "dtype": "float32",
4
  "num_hidden_layers": 32,
5
  "num_attention_heads": 20,
6
  "hidden_size": 1280,
 
1
  {
2
  "architecture": "WhisperEncoder",
3
+ "dtype": "float16",
4
  "num_hidden_layers": 32,
5
  "num_attention_heads": 20,
6
  "hidden_size": 1280,