Typhoon-Audio Preview

llama-3-typhoon-audio-8b-2411 is a 🇹🇭 Thai audio-language model. It supports both text and audio input modalities natively while the output is text. This version (November 2024) is our second iteration of the audio-language model after llama-3-typhoon-v1.5-8b-audio-preview.

More details can be found in our technical report. *To acknowledge Meta's effort in creating the foundation model and to comply with the license, we explicitly include "llama-3" in the model name.

Model Description

  • Model type: The LLM is based on TyphoonLLM, and the audio encoder is based on Whisper's encoder and BEATs.
  • Requirement: transformers 4.38.0 or newer.
  • Primary Language(s): Thai 🇹🇭 and English 🇬🇧
  • Demo: https://audio.opentyphoon.ai/
  • License: Llama 3 Community License

Usage Example

from transformers import AutoModel
import soundfile as sf
import librosa

# Initialize from the trained model
model = AutoModel.from_pretrained(
    "scb10x/llama-3-typhoon-audio-8b-2411", 
    torch_dtype=torch.float16,
    trust_remote_code=True
)
model.to("cuda")
model.eval()

# read a wav file (it needs to be in 16 kHz and clipped to 30 seconds)
audio, sr = sf.read("path_to_your_audio.wav")
if len(audio.shape) == 2:
    audio = audio[:, 0]
if len(audio) > 30 * sr:
    audio = audio[: 30 * sr]
if sr != 16000:
    audio = librosa.resample(audio, orig_sr=sr, target_sr=16000, res_type="fft")

# Run generation
prompt_pattern="<|start_header_id|>system<|end_header_id|>\n\nYou are a helpful assistant. Your name is Typhoon.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n<Speech><SpeechHere></Speech> {}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
response = model.generate(
    audio=audio,
    prompt="transcribe this audio",
    prompt_pattern=prompt_pattern,
    do_sample=False,
    max_new_tokens=512,
    repetition_penalty=1.1,
    num_beams=1,
    # temperature=0.4,
    # top_p=0.9,
)
print(response)

Generation Parameters:

  • audio -- audio input, e.g., using soundfile.read or librosa.resample to read a wav file like the example above
  • prompt (str) -- Text input to the model
  • prompt_pattern (str) -- Chat template that is augmented with special tokens, and it must be set the same as one during training
  • max_new_tokens (int, optional, defaults to 1024)
  • num_beams (int, optional, defaults to 4)
  • do_sample (bool, optional, defaults to True)
  • top_p (float, optional, defaults to 0.9)
  • repetition_penalty (float, optional, defaults to 1.0),
  • length_penalty (float, optional, defaults to 1.0),
  • temperature (float, optional, defaults to 1.0),

This is also model.generate_stream() for streaming generation. Please refer to modeling_typhoonaudio.py for this function.

Evaluation Results

More information is provided in our technical report.

Model ASR-en (WER↓) ASR-th (WER↓) En2Th (BLEU↑) X2Th (BLEU↑) Th2En (BLEU↑)
SALMONN-13B 5.79 98.07 0.07 0.10 14.97
DiVA-8B 30.28 65.21 9.82 5.31 7.97
Gemini-1.5-pro-001 5.98 13.56 20.69 13.52 22.54
Typhoon-Audio-Preview 8.72 14.17 17.52 10.67 24.14
Typhoon-Audio-2411 5.83 14.04 27.15 15.93 33.25
Model Gender-th (Acc) SpokenQA-th (F1) SpeechInstruct-(en,th)
SALMONN-13B 93.26 2.95 2.47, 1.18
DiVA-8B 50.12 15.13 6.81, 2.68
Gemini-1.5-pro-001 81.32 62.10 3.24, 3.93
Typhoon-Audio-Preview 93.74 64.60 5.62, 6.11
Typhoon-Audio-2411 75.65 70.01 6.00, 6.79

Intended Uses & Limitations

This model is experimental and may not always follow human instructions accurately, making it prone to generating hallucinations. Additionally, the model lacks moderation mechanisms and may produce harmful or inappropriate responses. Developers should carefully assess potential risks based on their specific applications.

Follow us & Support

Acknowledgements

We would like to thank the SALMONN team for open-sourcing their code and data, and thanks to the Biomedical and Data Lab at Mahidol University for releasing the fine-tuned Whisper that allowed us to adopt its encoder. Thanks to many other open-source projects for their useful knowledge sharing, data, code, and model weights.

Typhoon Team

Potsawee Manakul, Sittipong Sripaisarnmongkol, Natapong Nitarach, Warit Sirichotedumrong, Adisai Na-Thalang, Phatrasek Jirabovonvisut, Parinthapat Pengpun, Krisanapong Jirayoot, Pathomporn Chokchainant, Kasima Tharnpipitchai, Kunat Pipatanakul

Downloads last month
71
Inference Examples
Inference API (serverless) does not yet support model repos that contain custom code.