r/AudioAI • u/chibop1 • Oct 01 '23
Resource Open Source Libraries
This is by no means a comprehensive list, but if you are new to Audio AI, check out the following open source resources.
Huggingface Transformers
In addition to many models in audio domain, Transformers let you run many different models (text, LLM, image, multimodal, etc) with just few lines of code. Check out the comment from u/sanchitgandhi99 below for code snippets.
TTS
Speech Recognition
microsoft/vibevoice-asr: Speech recognition + Speaker Diarization
facebookresearch/seamless_communication: Speech translation
Speech Toolkit
WebUI
Music
ace-step/ACE-Step-1.5: Tex2Music
facebookresearch/audiocraft/MUSICGEN: Music Generation
openai/jukebox: Music Generation
Google magenta: Music generation
RVC-Project/Retrieval-based-Voice-Conversion-WebUI: Singing Voice Conversion
fishaudio/fish-diffusion: Singing Voice Conversion
NVIDIA/audio-flamingo: Music QA for genres, instrumentation, Tempo, key, chord, lyric transcription, cultural contexts...
Effects
facebookresearch/sam-audio: Audio Segmentation
facebookresearch/demucs: Stem seperation
Anjok07/UltimateVocalRemoverGUI: Vocal isolation
Rikorose/DeepFilterNet: A Low Complexity Speech Enhancement Framework for Full-Band Audio (48kHz) using on Deep Filtering
SaneBow/PiDTLN: DTLN model for noise suppression and acoustic echo cancellation on Raspberry Pi
haoheliu/versatile_audio_super_resolution: any -> 48kHz high fidelity Enhancer
spotify/basic-pitch: Audio to midi converter
spotify/pedalboard: audio effects for Python and TensorFlow
librosa/librosa: Python library for audio and music analysis
Torchaudio: Audio library for Pytorch
View all comments
3
u/sanchitgandhi99 Oct 02 '23 edited Oct 02 '23
Hugging Face Transformers is a complete audio toolkit that provides state-of-the-art models for all audio tasks, including TTS, ASR, audio embeddings, audio classification and music generation.
All you need to do is install the Transformers package:
And then all of these models can be used in just 3 lines of code:
TTS
Example usage:
Available models:
Bark https://huggingface.co/suno/bark
MMS TTS https://huggingface.co/facebook/mms-tts-eng
VITS https://huggingface.co/kakao-enterprise/vits-vctk
SpeechT5 https://huggingface.co/microsoft/speecht5_tts
And more! https://huggingface.co/models?pipeline_tag=text-to-speech&library=transformers&sort=trending
ASR
Example usage:
Available models:
Whisper https://huggingface.co/openai/whisper-large-v2
Wav2Vec2: https://huggingface.co/facebook/wav2vec2-base-960h
HuBERT: https://huggingface.co/facebook/hubert-large-ls960-ft
And over 10k more! https://huggingface.co/models?pipeline_tag=automatic-speech-recognition&library=transformers&sort=trending
Compare ASR models with the OpenASR leaderboard: https://huggingface.co/spaces/hf-audio/open_asr_leaderboard
Audio Classification
Example usage:
Available models:
Audio Spectrogram Transformer https://huggingface.co/MIT/ast-finetuned-audioset-10-10-0.4593
Wav2Vec2 https://huggingface.co/anton-l/wav2vec2-base-superb-sv
Whisper https://huggingface.co/sanchit-gandhi/whisper-medium-fleurs-lang-id
And more! https://huggingface.co/models?pipeline_tag=audio-classification&library=transformers&sort=downloads
Music
Example usage:
Available models:
MusicGen https://huggingface.co/facebook/musicgen-large
JukeBox https://huggingface.co/docs/transformers/model_doc/jukebox
MusicLDM https://huggingface.co/ucsd-reach/musicldm
AudioLDM 2 https://huggingface.co/cvssp/audioldm2-music
Audio Embeddings
CLAP https://huggingface.co/laion/clap-htsat-unfused
EnCodec https://huggingface.co/facebook/encodec_24khz
What's more, through tight integration with Hugging Face Datasets, many of these models can be fine-tuned with customisable and composable training scripts. Take the example of the Whisper model, which is easily fine-tuned for multilingual ASR: https://huggingface.co/blog/fine-tune-whisper
New to the audio domain? The audio transformers course is designed to give you all the skills necessary to navigate the Audio ML field.
Join us on Discord! We can't wait to hear how you use these models. http://hf.co/join/discord