Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
 
 ENV DEBIAN_FRONTEND=noninteractive
 
@@ -28,7 +28,7 @@ WORKDIR $HOME/app
 RUN git clone -b magnet https://github.com/camenduru/audiocraft $HOME/app 
     
 # Install dependencies
-RUN pip install -q av julius omegaconf einops num2words flashy torchaudio==2.0.2 torchmetrics sentencepiece gradio
+RUN pip install -q av julius omegaconf einops num2words flashy torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121 torchmetrics sentencepiece gradio
 RUN pip install -q https://github.com/camenduru/wheels/releases/download/colab/xformers-0.0.20+1dc3d7a.d20240115-cp310-cp310-linux_x86_64.whl
 RUN pip install -q .