Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -14,6 +14,6 @@ COPY stable_diffusion_engine.py demo.py demo_web.py /src/
|
|
14 |
COPY data/ /src/data/
|
15 |
|
16 |
# download models
|
17 |
-
RUN python3
|
18 |
|
19 |
-
ENTRYPOINT ["python3", "
|
|
|
14 |
COPY data/ /src/data/
|
15 |
|
16 |
# download models
|
17 |
+
RUN python3 demo.py --num-inference-steps 1 --prompt "test" --output /tmp/test.jpg
|
18 |
|
19 |
+
ENTRYPOINT ["python3", "demo.py"]
|