Update Dockerfile

Dockerfile

@@ -15,7 +15,7 @@ ENV HOME=/home/user
 ENV PATH=/home/user/.local/bin:$PATH
 WORKDIR /home/user/app
 
-RUN pip install --no-cache-dir addict albumentations addict albumentations basicsr blendmodes einops gradio imageio imageio-ffmpeg invisible-watermark kornia numba omegaconf open_clip_torch prettytable pytorch-lightning safetensors streamlit streamlit-drawable-canvas test-tube timm torchmetrics transformers webdataset yapf
+RUN pip install --no-cache-dir addict albumentations addict albumentations basicsr blendmodes einops gradio imageio imageio-ffmpeg invisible-watermark kornia numba omegaconf open_clip_torch prettytable pytorch-lightning safetensors streamlit streamlit-drawable-canvas test-tube timm torchmetrics transformers torch torchvision webdataset xformers yapf
 
 ENV PYTHONPATH=$HOME/app
 ENV PYTHONUNBUFFERED=1