Sem descrição

Lengyue 8f75f764ad Fix incorrect batch time há 2 anos atrás
preparing_data 9adf687fb1 Update base training code há 2 anos atrás
speech_lm 8f75f764ad Fix incorrect batch time há 2 anos atrás
.dockerignore 70797ef7d5 Add dockerignore há 2 anos atrás
.gitignore 7650b2ac43 Update much robust dataloader há 2 anos atrás
.pre-commit-config.yaml e4ce816175 Update deps há 2 anos atrás
README.md 3fc0a688ca Update Readme há 2 anos atrás
dockerfile 7650b2ac43 Update much robust dataloader há 2 anos atrás
requirements.txt 2733ed600f Accelerate training with fsdp há 2 anos atrás
setup.py 7650b2ac43 Update much robust dataloader há 2 anos atrás
train.sh a0a5c3e95c Update training script há 2 anos atrás

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 .