Update env.sh (#73)

specified accelerate version to fix compatibility issue
open-mmlab · Dec 29, 2023 · d34eaa3 · d34eaa3
1 parent cae20ce
commit d34eaa3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/env.sh b/env.sh
@@ -9,7 +9,7 @@ conda install -c conda-forge ffmpeg
 # Pip packages
 pip install setuptools ruamel.yaml tqdm colorama easydict tabulate loguru json5 Cython unidecode inflect argparse g2p_en tgt librosa matplotlib typeguard einops omegaconf hydra-core humanfriendly pandas
 
-pip install tensorboard tensorboardX torch==2.0.1 torchaudio==2.0.2 torchvision==0.15.2 transformers diffusers accelerate praat-parselmouth audiomentations pedalboard ffmpeg-python==0.2.0 pyworld diffsptk nnAudio unidecode inflect ptwt
+pip install tensorboard tensorboardX torch==2.0.1 torchaudio==2.0.2 torchvision==0.15.2 accelerate==0.24.1 transformers diffusers praat-parselmouth audiomentations pedalboard ffmpeg-python==0.2.0 pyworld diffsptk nnAudio unidecode inflect ptwt
 
 pip install torchmetrics pymcd openai-whisper frechet_audio_distance asteroid