mudler commited on
Commit
9bfac15
·
verified ·
1 Parent(s): 21e89ac

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +63 -0
README.md ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: it
3
+ license: mit
4
+ tags:
5
+ - whisper
6
+ - automatic-speech-recognition
7
+ - italian
8
+ - localai
9
+ datasets:
10
+ - mozilla-foundation/common_voice_25_0
11
+ base_model: openai/whisper-small
12
+ pipeline_tag: automatic-speech-recognition
13
+ ---
14
+
15
+ # whisper-small-it
16
+
17
+ Fine-tuned [openai/whisper-small](https://huggingface.co/openai/whisper-small) (244M params) for Italian ASR.
18
+
19
+ **Author:** Ettore Di Giacinto
20
+
21
+ Brought to you by the [LocalAI](https://github.com/mudler/LocalAI) team. This model can be used directly with [LocalAI](https://localai.io).
22
+
23
+ ## Results
24
+
25
+ Evaluated on Common Voice 25.0 Italian test set (15,184 samples):
26
+
27
+ | Step | WER |
28
+ |------|-----|
29
+ | 1000 | 18.36% |
30
+ | 3000 | 15.45% |
31
+ | 5000 | 14.58% |
32
+ | 7000 | 13.61% |
33
+ | 10000 | **13.0%** |
34
+
35
+ ## Training Details
36
+
37
+ - **Base model:** openai/whisper-small (244M parameters)
38
+ - **Dataset:** Common Voice 25.0 Italian (173k train, 15k dev, 15k test)
39
+ - **Steps:** 10,000
40
+ - **Precision:** bf16 on NVIDIA GB10
41
+
42
+ ## Usage
43
+
44
+ ### Transformers
45
+
46
+ ```python
47
+ from transformers import pipeline
48
+
49
+ pipe = pipeline("automatic-speech-recognition", model="LocalAI-io/whisper-small-it")
50
+ result = pipe("audio.mp3", generate_kwargs={"language": "it", "task": "transcribe"})
51
+ print(result["text"])
52
+ ```
53
+
54
+ ### CTranslate2 / faster-whisper
55
+
56
+ For optimized CPU inference: [LocalAI-io/whisper-small-it-ct2-int8](https://huggingface.co/LocalAI-io/whisper-small-it-ct2-int8)
57
+
58
+ ## Links
59
+
60
+ - **Multi-dataset version:** [LocalAI-io/whisper-small-it-multi](https://huggingface.co/LocalAI-io/whisper-small-it-multi) (WER 15.6%)
61
+ - **CTranslate2 INT8:** [LocalAI-io/whisper-small-it-ct2-int8](https://huggingface.co/LocalAI-io/whisper-small-it-ct2-int8)
62
+ - **Code:** [github.com/localai-org/whisper-it](https://github.com/localai-org/whisper-it)
63
+ - **LocalAI:** [github.com/mudler/LocalAI](https://github.com/mudler/LocalAI)