Hugging Face's logo

Steveeeeeeen
/
whisper-medium-en-trt-llm-checkpoint

Model card Files Community
Steveeeeeeen HF staff commited on
Commit
ae0eb5d
·
1 Parent(s): b9e5bce

Change weight to float32

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": "float16",
4
- "logits_dtype": "float16",
5
  "num_hidden_layers": 24,
6
  "num_attention_heads": 16,
7
  "hidden_size": 1024,
 
1
  {
2
  "architecture": "DecoderModel",
3
+ "dtype": "float32",
4
+ "logits_dtype": "float32",
5
  "num_hidden_layers": 24,
6
  "num_attention_heads": 16,
7
  "hidden_size": 1024,
encoder/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architecture": "WhisperEncoder",
3
- "dtype": "float16",
4
  "num_hidden_layers": 24,
5
  "num_attention_heads": 16,
6
  "hidden_size": 1024,
 
1
  {
2
  "architecture": "WhisperEncoder",
3
+ "dtype": "float32",
4
  "num_hidden_layers": 24,
5
  "num_attention_heads": 16,
6
  "hidden_size": 1024,