Aucune description

Lengyue e4839d6e84 Use 4 workers il y a 2 ans
fish_speech e4839d6e84 Use 4 workers il y a 2 ans
tools d16c9cb695 Update huggingface dataset builder il y a 2 ans
.dockerignore 0c084ec779 Apply gradient accumulate to save memory il y a 2 ans
.gitignore 6d0e669f8c Rename project to fish-speech il y a 2 ans
.pre-commit-config.yaml e4ce816175 Update deps il y a 2 ans
README.md 3fc0a688ca Update Readme il y a 2 ans
dockerfile 9920873cf2 Optimize schedule il y a 2 ans
pyrightconfig.json f7f2c03282 Support pytorch lightning il y a 2 ans
requirements.txt 23109ecb94 Optimize training loop & upgrade transformers il y a 2 ans
setup.py 6d0e669f8c Rename project to fish-speech il y a 2 ans

README.md

Speech LLM

Setup

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

# Install requirements
pip3 install -r requirements.txt

# Install flash-attn
MAX_JOBS=4 pip install flash-attn --no-build-isolation

# Install speech-llm
pip3 install -e .