Xenova HF Staff commited on
Commit
e2b75e8
·
verified ·
1 Parent(s): a38d08a

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -0
README.md ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ {}
3
+ ---
4
+ Code to generate:
5
+
6
+ ```py
7
+ from transformers import WhisperForConditionalGeneration, AutoProcessor
8
+
9
+ new_config_values = dict(
10
+ d_model = 16,
11
+ decoder_attention_heads = 4,
12
+ decoder_layers = 1,
13
+ encoder_attention_heads = 4,
14
+ encoder_layers = 1,
15
+ num_hidden_layers = 1,
16
+
17
+ ignore_mismatched_sizes=True,
18
+ )
19
+ original_model = WhisperForConditionalGeneration.from_pretrained('openai/whisper-tiny', **new_config_values)
20
+ original_model.save_pretrained('converted')
21
+
22
+ original_processor = AutoProcessor.from_pretrained('openai/whisper-tiny')
23
+ original_processor.save_pretrained('converted')
24
+ ```
25
+
26
+ Followed by:
27
+ ```sh
28
+ $ mkdir -p ./converted/onnx
29
+ $ optimum-cli export onnx -m ./converted ./converted/onnx --task automatic-speech-recognition-with-past
30
+ $ find ./converted/onnx -type f ! -name "*.onnx" -delete
31
+ ```