Hugging Face's logo

Spaces:
lemonteaa
/
fastsdcpu-test
Sleeping

App Files Community
lemonteaa commited on
Commit
decfbe4
·
1 Parent(s): fed5ca0

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -3,6 +3,7 @@ FROM python:3
3
  WORKDIR /code
4
 
5
  RUN apt-get update && apt-get install -y git
 
6
  RUN git clone https://github.com/rupeshs/fastsdcpu && chmod a+x ./fastsdcpu/install.sh && chmod a+x ./fastsdcpu/start-webui.sh && /code/fastsdcpu/install.sh
7
 
8
  WORKDIR /code/fastsdcpu
 
3
  WORKDIR /code
4
 
5
  RUN apt-get update && apt-get install -y git
6
+ RUN pip3 install torch==2.0.1 --index-url https://download.pytorch.org/whl/cpu
7
  RUN git clone https://github.com/rupeshs/fastsdcpu && chmod a+x ./fastsdcpu/install.sh && chmod a+x ./fastsdcpu/start-webui.sh && /code/fastsdcpu/install.sh
8
 
9
  WORKDIR /code/fastsdcpu