Views
No views yet
bosonai/higgs-audio-understanding-v3-1.7b (checkpoint-65000) with a
LoRA (rank=64, alpha=128) merged into the base weights. The LoRA was trained
on a curated ASR mix (AMI IHM train, Earnings22 skipped, GigaSpeech XS train,
LibriSpeech train.100 + train.500, SpgiSpeech S train, TEDLium train, VoxPopuli-en
train). No ESB test data was used in training.| Dataset | WER (%) |
|---|---|
| AMI | 10.03 |
| Earnings-22 | 8.95 |
| GigaSpeech | 8.16 |
| LibriSpeech clean | 1.39 |
| LibriSpeech other | 2.80 |
| SPGISpeech | 3.76 |
| TED-LIUM | 2.76 |
| VoxPopuli | 6.07 |
| Macro avg | 5.49 |
run_eval_higgs_audio.py
script, max_new_tokens=1024, greedy decoding, Whisper English normalizer.1from transformers import AutoModel, AutoTokenizer
2import torch, numpy as np, soundfile as sf
3
4model = AutoModel.from_pretrained(
5 "bosonai/higgs-audio-v3-stt-v2",
6 torch_dtype=torch.bfloat16,
7 trust_remote_code=True,
8 attn_implementation="eager",
9 device_map="cuda:0",
10)
11tok = AutoTokenizer.from_pretrained("bosonai/higgs-audio-v3-stt-v2")
12model.audio_out_bos_token_id = tok.convert_tokens_to_ids("<|audio_out_bos|>")
13model.audio_eos_token_id = tok.convert_tokens_to_ids("<|audio_eos|>")
14
15# Load bundled transcribe.py
16from transformers.utils import cached_file
17import runpy, os, sys
18path = cached_file("bosonai/higgs-audio-v3-stt-v2", "transcribe.py")
19for f in ["higgs_audio_collator.py","modeling_higgs_audio_xcodec.py","utils.py","common.py","configuration_higgs_audio.py"]:
20 cached_file("bosonai/higgs-audio-v3-stt-v2", f)
21sys.path.insert(0, os.path.dirname(path))
22transcribe_batch = runpy.run_path(path)["transcribe_batch"]
23
24audio, sr = sf.read("example.wav")
25print(transcribe_batch(model, tok, [audio.astype(np.float32)], sample_rates=sr))1git clone https://github.com/huggingface/open_asr_leaderboard
2cd open_asr_leaderboard
3python run_eval_higgs_audio.py \
4 --model_id bosonai/higgs-audio-v3-stt-v2 \
5 --dataset_path hf-audio/open-asr-leaderboard-sorted \
6 --dataset ami --split test --device 0 --batch_size 4