SAVRN
Search Contact SAVRN

Open-weight model · Audio classification

ast-finetuned-speech-commands-v2

by Massachusetts Institute of Technology MIT/ast-finetuned-speech-commands-v2

Audio Spectrogram Transformer (AST) model fine-tuned on Speech Commands v2. It was introduced in the paper AST: Audio Spectrogram Transformer by Gong et al. and first released in this repository.

Parameters85M
Context
Weights683.3 MB
Licensebsd-3-clause
AccessOpen weights
Monthly Downloads7.8k

Runs On

What it takes to serve ast-finetuned-speech-commands-v2 (85M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.

PrecisionWeightsMemory neededCheapest setupPer hourAlso fits
16-bit 0.2 GB 0.2 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
8-bit 0.1 GB 0.1 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
4-bit 0.0 GB 0.1 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00

Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Sep 18, 2026.

Model Card

By Massachusetts Institute of Technology, published under bsd-3-clause, revision 315b0b847a3c.

Audio Spectrogram Transformer (AST) model fine-tuned on Speech Commands v2. It was introduced in the paper AST: Audio Spectrogram Transformer by Gong et al. and first released in this repository. Disclaimer: The team releasing Audio Spectrogram Transformer did not write a model card for this model so this model card has been written by the Hugging Face team. The Audio Spectrogram Transformer is equivalent to ViT, but applied on audio. Audio is first turned into an image (as a spectrogram), after which a Vision Transformer is applied. The model gets state-of-the-art results on several audio classification benchmarks. You can use the raw model for classifying audio into one of the Speech…

Read Massachusetts Institute of Technology's full model card

Audio Spectrogram Transformer (fine-tuned on Speech Commands v2)

Audio Spectrogram Transformer (AST) model fine-tuned on Speech Commands v2. It was introduced in the paper AST: Audio Spectrogram Transformer by Gong et al. and first released in this repository.

Disclaimer: The team releasing Audio Spectrogram Transformer did not write a model card for this model so this model card has been written by the Hugging Face team.

Model description

The Audio Spectrogram Transformer is equivalent to ViT, but applied on audio. Audio is first turned into an image (as a spectrogram), after which a Vision Transformer is applied. The model gets state-of-the-art results on several audio classification benchmarks.

Usage

You can use the raw model for classifying audio into one of the Speech Commands v2 classes. See the documentation for more info.

Configuration

Architecture
ASTForAudioClassification
Layers
12
Hidden size
768
Feed-forward size
3,072
Attention heads
12
Stored precision
float32
Model type
audio-spectrogram-transformer

Identity and Version

Repository
MIT/ast-finetuned-speech-commands-v2
Publisher
Massachusetts Institute of Technology
Task
Audio classification
Modality
Audio
Library
transformers
Parameters
85M parameters
Languages
Not stated by the source
Revision
315b0b847a3ca207e68b718503ad72066612eacd
First published
2022-11-14
Last updated
2023-09-10

Files and Weights

6 files, 683.3 MB in total. The weights are 2 files totalling 683.3 MB in bin, safetensors.

Weights2 files · 683.3 MB
Configuration2 files · 2.1 KB
Documentation1 file · 1.4 KB
Repository1 file · 1.5 KB
Every file
FileTypeSizeSHA-256
model.safetensorsWeights341.6 MB bde044423042
pytorch_model.binWeights341.7 MB 26cce475c88e
config.jsonConfiguration1.8 KB
preprocessor_config.jsonConfiguration295 B
README.mdDocumentation1.4 KB
.gitattributesRepository1.5 KB

License and Download

License
bsd-3-clause
Access
Open weights, no gate
Download size
683.3 MB
Download from Massachusetts Institute of Technology

Released by Massachusetts Institute of Technology through its official repository on Hugging Face. Read the license.

Built From

Evaluations

Each result is shown as reported, with the conditions its reporter stated. None is a SAVRN measurement. A comparison lines two results up only when their configuration, unit and setup are all stated and identical.

BenchmarkConditionsResultReported byRevisionDate
Speech Commands v2 Task audio-classificationMetric accuracyComparison conditions not established 98.12 MIT
Publisher reported
Evaluated revision not stated

Memory Requirements

PrecisionWeights in memory
As published683.3 MB
16-bit0.2 GB
8-bit0.1 GB
4-bit0.0 GB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About ast-finetuned-speech-commands-v2

How much GPU memory does ast-finetuned-speech-commands-v2 need?

About 0.2 GB at 16-bit and 0.1 GB at 4-bit: the weights (85M parameters) plus a working margin. A long context needs more.

What is the cheapest GPU to run ast-finetuned-speech-commands-v2 on?

At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.

Can I use ast-finetuned-speech-commands-v2 commercially?

Yes. ast-finetuned-speech-commands-v2 is released under BSD 3-Clause License. The BSD 3-Clause License is permissive. It permits commercial use and redistribution with the copyright notice, and forbids using the authors' names to endorse derived products without permission.

Similar Models

Model · Audio classification

dasheng-base

Speech Team, Xiaomi MiLM Plus

Dasheng (Deep Audio-Signal Holistic Embeddings), or “大声” ("great sound"), is a general-purpose audio encoder trained on a large-scale self-supervised learning task. Dasheng is designed to capture rich audio information across various domains, including speech, music, and environmental sounds. The model is trained on 272,356 hours of diverse audio data with 1.2 billion parameters, and exhibits significant performance gains on the HEAR benchmark. Dasheng outperforms previous works on CREMA-D, LibriCount, Speech Commands, VoxLingua, and competes well in music and environmental sound classification tasks. examplefinetuneesc50.ipynb demonstrates how to train a linear head on the ESC-50 dataset…

Open weights apache-2.0 85M parameters transformers

Model · Audio classification

ced-base

Speech Team, Xiaomi MiLM Plus

CED are simple ViT-Transformer-based models for audio tagging, achieving sota performance on Audioset. Notable differences from other available models include: 1. Simplification for finetuning: Batchnormalization of Mel-Spectrograms. During finetuning one does not need to first compute mean/variance over the dataset, which is common for AST. 1. Support for variable length inputs. Most other models use a static time-frequency position embedding, which hinders the model's generalization to segments shorter than 10s. Many previous transformers simply pad their input to 10s in order to avoid the performance impact, which in turn slows down training/inference drastically. 1. Training/Inference…

Open weights apache-2.0 86M parameters transformers

MAEST is a family of Transformer models based on PASST and focused on music analysis applications. The MAEST models are also available for inference in the Essentia library and for inference and training in the official repository. You can try the MAEST interactive demo on replicate. MAEST is a music audio representation model pre-trained on the task of music style classification. According to the evaluation reported in the original paper, it reports good performance in several downstream music analysis tasks. The MAEST models can make predictions for a taxonomy of 400 music styles derived from the public metadata of Discogs. The MAEST models have reported good performance in downstream…

Open weights cc-by-nc-sa-4.0 86M parameters transformers

Audio Spectrogram Transformer (AST) model fine-tuned on AudioSet. It was introduced in the paper AST: Audio Spectrogram Transformer by Gong et al. and first released in this repository. Disclaimer: The team releasing Audio Spectrogram Transformer did not write a model card for this model so this model card has been written by the Hugging Face team. The Audio Spectrogram Transformer is equivalent to ViT, but applied on audio. Audio is first turned into an image (as a spectrogram), after which a Vision Transformer is applied. The model gets state-of-the-art results on several audio classification benchmarks. You can use the raw model for classifying audio into one of the AudioSet classes. See…

Open weights bsd-3-clause 86M parameters transformers

Audio Spectrogram Transformer (AST) model fine-tuned on AudioSet. It was introduced in the paper AST: Audio Spectrogram Transformer by Gong et al. and first released in this repository. Disclaimer: The team releasing Audio Spectrogram Transformer did not write a model card for this model so this model card has been written by the Hugging Face team. The Audio Spectrogram Transformer is equivalent to ViT, but applied on audio. Audio is first turned into an image (as a spectrogram), after which a Vision Transformer is applied. The model gets state-of-the-art results on several audio classification benchmarks. You can use the raw model for classifying audio into one of the AudioSet classes. See…

Open weights bsd-3-clause 87M parameters transformers

The model expects a raw audio signal as input and outputs predictions for age in a range of approximately 0...1 (0...100 years) and gender expressing the probababilty for being child, female, or male. In addition, it also provides the pooled states of the last transformer layer. The model was created by fine-tuning Wav2Vec2-Large-Robust Timit and For this version of the model we only trained the first six transformer layers. An ONNX export of the model is available from Further details are given in the associated paper and tutorial.

Open weights cc-by-nc-sa-4.0 91M parameters transformers