pere commited on
Commit
7307350
1 Parent(s): 0c591d9

PyTorch models converted

Browse files
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "openai/whisper-medium",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
@@ -146,7 +146,7 @@
146
  50362
147
  ],
148
  "torch_dtype": "float32",
149
- "transformers_version": "4.31.0.dev0",
150
  "use_cache": true,
151
  "use_weighted_layer_sum": false,
152
  "vocab_size": 51865
 
1
  {
2
+ "_name_or_path": ".",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
 
146
  50362
147
  ],
148
  "torch_dtype": "float32",
149
+ "transformers_version": "4.30.1",
150
  "use_cache": true,
151
  "use_weighted_layer_sum": false,
152
  "vocab_size": 51865
convert_to_pytorch.py ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ from transformers import WhisperForConditionalGeneration
2
+ model = WhisperForConditionalGeneration.from_pretrained(".", from_flax=True)
3
+ model.save_pretrained(".")
4
+ print("\nModel is now converted to PyTorch. There should be a new pytorch_model.bin-file in this directory.")
5
+
generation_config.json CHANGED
@@ -220,5 +220,5 @@
220
  "transcribe": 50359,
221
  "translate": 50358
222
  },
223
- "transformers_version": "4.31.0.dev0"
224
  }
 
220
  "transcribe": 50359,
221
  "translate": 50358
222
  },
223
+ "transformers_version": "4.30.1"
224
  }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d556f5a1c038eee4ec61beb097a4b668a65524ba2772778cc0b80d123d9ad32f
3
+ size 3055741593