darkstorm2150 commited on
Commit
983ac98
1 Parent(s): 3a5fd8b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -14,6 +14,7 @@ WORKDIR /content
14
  USER user
15
 
16
  RUN pip3 install --upgrade pip
 
17
  RUN pip install https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp310-cp310-linux_x86_64.whl
18
  RUN pip install --pre triton
19
  RUN pip install numexpr
 
14
  USER user
15
 
16
  RUN pip3 install --upgrade pip
17
+ RUN pip install torchmetrics==0.11.4
18
  RUN pip install https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp310-cp310-linux_x86_64.whl
19
  RUN pip install --pre triton
20
  RUN pip install numexpr