Skip to content

Introduction

Warning

We assume no responsibility for any illegal use of the codebase. Please refer to the local laws regarding DMCA (Digital Millennium Copyright Act) and other relevant laws in your area.
This codebase and all models are released under the CC-BY-NC-SA-4.0 license.

Requirements

  • GPU Memory: 4GB (for inference), 8GB (for fine-tuning)
  • System: Linux, Windows

Windows Setup

Attention

We strongly suggest non-professional windows users use our official GUI to run the project. GUI is here.

Professional Windows users may consider using WSL2 or Docker to run the codebase.

# Create a python 3.10 virtual environment, you can also use virtualenv
conda create -n fish-speech python=3.10
conda activate fish-speech

# Install pytorch
pip3 install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu121

# Install fish-speech
pip3 install -e .

# (Enable acceleration) Install triton-windows
pip install https://github.com/AnyaCoder/fish-speech/releases/download/v0.1.0/triton_windows-0.1.0-py3-none-any.whl

Linux Setup

See pyproject.toml for details.

# Create a python 3.10 virtual environment, you can also use virtualenv
conda create -n fish-speech python=3.10
conda activate fish-speech

# Install pytorch
pip3 install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1

# (Ubuntu / Debian User) Install sox + ffmpeg
apt install libsox-dev ffmpeg 

# (Ubuntu / Debian User) Install pyaudio 
apt install build-essential \
    cmake \
    libasound-dev \
    portaudio19-dev \
    libportaudio2 \
    libportaudiocpp0

# Install fish-speech
pip3 install -e .[stable]

MacOS setup

If you want to perform inference on MPS, please add the --device mps flag. Please refer to this PR for a comparison of inference speeds.

Warning

The compile option is not officially supported on Apple Silicon devices, so there is no guarantee that inference speed will improve.

# install dependencies
brew install portaudio
# create a python 3.10 virtual environment, you can also use virtualenv
conda create -n fish-speech python=3.10
conda activate fish-speech
# install pytorch
pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1
# install fish-speech
pip install -e .[stable]

Docker Setup

  1. Install NVIDIA Container Toolkit:

    To use GPU for model training and inference in Docker, you need to install NVIDIA Container Toolkit:

    For Ubuntu users:

    # Add repository
    curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
        && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
            sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
            sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
    # Install nvidia-container-toolkit
    sudo apt-get update
    sudo apt-get install -y nvidia-container-toolkit
    # Restart Docker service
    sudo systemctl restart docker
    

    For users of other Linux distributions, please refer to: NVIDIA Container Toolkit Install-guide.

  2. Pull and run the fish-speech image

    # Pull the image
    docker pull fishaudio/fish-speech:latest-dev
    # Run the image
    docker run -it \
        --name fish-speech \
        --gpus all \
        -p 7860:7860 \
        fishaudio/fish-speech:latest-dev \
        zsh
    # If you need to use a different port, please modify the -p parameter to YourPort:7860
    
  3. Download model dependencies

    Make sure you are in the terminal inside the docker container, then download the required vqgan and llama models from our huggingface repository.

    huggingface-cli download fishaudio/fish-speech-1.5 --local-dir checkpoints/fish-speech-1.5
    
  4. Configure environment variables and access WebUI

    In the terminal inside the docker container, enter export GRADIO_SERVER_NAME="0.0.0.0" to allow external access to the gradio service inside docker. Then in the terminal inside the docker container, enter python tools/run_webui.py to start the WebUI service.

    If you're using WSL or MacOS, visit http://localhost:7860 to open the WebUI interface.

    If it's deployed on a server, replace localhost with your server's IP.

Changelog

  • 2024/12/03: Updated Fish-Speech to 1.5 version, supports more languages, and reaches SOTA in the Open-Source field.
  • 2024/09/10: Updated Fish-Speech to 1.4 version, with an increase in dataset size and a change in the quantizer's n_groups from 4 to 8.
  • 2024/07/02: Updated Fish-Speech to 1.2 version, remove VITS Decoder, and greatly enhanced zero-shot ability.
  • 2024/05/10: Updated Fish-Speech to 1.1 version, implement VITS decoder to reduce WER and improve timbre similarity.
  • 2024/04/22: Finished Fish-Speech 1.0 version, significantly modified VQGAN and LLAMA models.
  • 2023/12/28: Added lora fine-tuning support.
  • 2023/12/27: Add gradient checkpointing, causual sampling, and flash-attn support.
  • 2023/12/19: Updated webui and HTTP API.
  • 2023/12/18: Updated fine-tuning documentation and related examples.
  • 2023/12/17: Updated text2semantic model, supporting phoneme-free mode.
  • 2023/12/13: Beta version released, includes VQGAN model and a language model based on LLAMA (phoneme support only).

Acknowledgements