Hugging Face's logo

Spaces:
lemonteaa
/
fastsdcpu-test
Sleeping

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

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,9 +1,9 @@
1
- FROM python:3
2
 
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
 
1
+ FROM cnstark/pytorch:2.0.1-py3.10.11-ubuntu22.04
2
 
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