Update Dockerfile

Dockerfile

@@ -21,8 +21,11 @@ WORKDIR $HOME/app
 # Clone your repository or add your code to the container
 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 torchtext==0.10.0 pytorch-lightning==1.4.2
+
 # Install Python dependencies
-RUN pip install -q xformers==0.0.16 pytorch_lightning==1.4.2 einops open-clip-torch omegaconf torchmetrics==0.6.0 triton opencv-python-headless scipy matplotlib lpips gradio chardet transformers facexlib
+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
 
 # Install open_clip from GitHub
 RUN pip install -q git+https://github.com/mlfoundations/[email protected]