pip install pydantic~=1.0

Dockerfile

@@ -33,7 +33,7 @@ RUN git clone -b main https://github.com/fffiloni/DiffBIR $HOME/app
 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
+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 pydantic~=1.0
 
 # Install open_clip from GitHub
 RUN pip install -q git+https://github.com/mlfoundations/[email protected]