Spaces:
Sleeping
Sleeping
use gpu img
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
|
2 |
-
FROM ubuntu:22.04
|
3 |
|
4 |
ARG LIVE_XL_TAR=live_xl_rel_ubuntu-22.04_202403190935.tar
|
5 |
|
@@ -18,5 +18,6 @@ WORKDIR /web-app
|
|
18 |
RUN wget https://github.com/jackalcooper/live_xl/releases/download/latest/${LIVE_XL_TAR}
|
19 |
RUN tar xf ${LIVE_XL_TAR}
|
20 |
ENV PORT=7860 SECRET_KEY_BASE="Z/VeN8hw0cQruiL3YyWG52IPq+hnG9Ucj1RuW0263Iddmd8f+j7i/b7NCs1fiLCx"
|
|
|
21 |
|
22 |
CMD [ "live_xl/bin/server", "start" ]
|
|
|
1 |
+
FROM oneflowinc/onediff:cu118
|
2 |
+
# FROM ubuntu:22.04
|
3 |
|
4 |
ARG LIVE_XL_TAR=live_xl_rel_ubuntu-22.04_202403190935.tar
|
5 |
|
|
|
18 |
RUN wget https://github.com/jackalcooper/live_xl/releases/download/latest/${LIVE_XL_TAR}
|
19 |
RUN tar xf ${LIVE_XL_TAR}
|
20 |
ENV PORT=7860 SECRET_KEY_BASE="Z/VeN8hw0cQruiL3YyWG52IPq+hnG9Ucj1RuW0263Iddmd8f+j7i/b7NCs1fiLCx"
|
21 |
+
ENV LIVE_XL_LIGHTNING_ARGS="--base stabilityai/stable-diffusion-xl-base-1.0 --repo ByteDance/SDXL-Lightning --cpkt sdxl_lightning_2step_unet.safetensors"
|
22 |
|
23 |
CMD [ "live_xl/bin/server", "start" ]
|