Update Dockerfile

Dockerfile

@@ -22,7 +22,7 @@ WORKDIR $HOME/app
 RUN git clone -b dev https://github.com/camenduru/DiffBIR $HOME/app
 
 # Install specific versions of PyTorch, TorchText, and PyTorch Lightning
-RUN pip install torch==1.12.1 torchvision==0.13.1 pytorch-lightning==1.4.2
+RUN pip install torch==1.12.1 torchvision==0.13.1 pytorch-lightning
 
 # Install Python dependencies
 RUN pip install -q xformers==0.0.16 einops open-clip-torch omegaconf torchmetrics==0.6.0 triton opencv-python-headless scipy matplotlib lpips gradio chardet transformers facexlib