Bez popisu

Lengyue 9ac8edef1e Support mix codebook training před 2 roky
data_server 57755e1e74 Add rust data server před 2 roky
fish_speech 9ac8edef1e Support mix codebook training před 2 roky
tools 9ac8edef1e Support mix codebook training před 2 roky
.dockerignore 57755e1e74 Add rust data server před 2 roky
.gitignore 57755e1e74 Add rust data server před 2 roky
.pre-commit-config.yaml f1ea577c4c [pre-commit.ci] pre-commit autoupdate před 2 roky
LICENSE 62710e34a4 Add llama inference tool chain před 2 roky
README.md 57755e1e74 Add rust data server před 2 roky
dockerfile f8cfbb4ac0 Greatly improved data loading speed před 2 roky
pyproject.toml 57755e1e74 Add rust data server před 2 roky
pyrightconfig.json f7f2c03282 Support pytorch lightning před 2 roky

README.md

Fish Speech

This repo is still under construction. Please check back later.

Setup

# Basic environment setup
conda create -n fish-speech python=3.10
conda activate fish-speech
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

# Install flash-attn (for linux)
pip3 install ninja && MAX_JOBS=4 pip3 install flash-attn --no-build-isolation

# Install fish-speech
pip3 install -e .

Rust Data Server

Since loading and shuffle the dataset is very slow and memory consuming, we use a rust server to load and shuffle the dataset. The server is based on GRPC and can be installed by

cd data_server
cargo build --release

Credits