Update Dockerfile

Dockerfile

@@ -3,6 +3,9 @@ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
 
 ENV DEBIAN_FRONTEND=noninteractive 
 
+# Install Python 3.8 (or any other version you need)
+RUN apt-get install -y python3.8
+
 RUN apt-get update -y && apt-get upgrade -y && apt-get install -y sudo && apt-get install -y python3-pip && pip3 install --upgrade pip
 RUN apt-get install -y curl gnupg wget htop sudo git git-lfs software-properties-common build-essential libgl1 zip unzip
 
@@ -25,7 +28,7 @@ RUN git clone -b dev https://github.com/camenduru/DiffBIR $HOME/app
 #RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
 
 # Install Python dependencies
-RUN pip3 install -q einops pytorch_lightning gradio omegaconf xformers==0.0.13 transformers lpips opencv-python
+RUN pip3 install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 transformers lpips opencv-python
 
 # Install open_clip from GitHub
 RUN pip3 install -q git+https://github.com/mlfoundations/open_clip@v2.20.0