hft: add tensorboard to Dockerfile
@@ -187,6 +187,7 @@ RUN pip install --upgrade pip
RUN pip install transformers==4.6.0 datasets jiwer==2.2.0 lang-trans==0.6.0 librosa==0.8.0
# Set torch version for CUDA 11
RUN pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
+RUN pip install tensorboard==2.7.0
# Cache the pretrained models
COPY download_wav2vec2.py /root/download_wav2vec2.py
RUN python /root/download_wav2vec2.py
