J5Tsai commited on
Commit
e148c21
·
1 Parent(s): 42da31c

use gpu img

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -1,5 +1,5 @@
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,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" ]