File tree Expand file tree Collapse file tree 2 files changed +23
-1
lines changed
Expand file tree Collapse file tree 2 files changed +23
-1
lines changed Original file line number Diff line number Diff line change @@ -144,6 +144,28 @@ function build_tensorrt_models() {
144144 echo " failed ComfyUI build_trt.py"
145145 exit 1
146146 )
147+
148+ # Dreamshaper-8-Dmd-1kstep static dynamic 488x704
149+ # TODO(pschroedl): Remove the script download with curl.
150+ # It will be present in the base image once https://github.com/livepeer/comfystream/pull/107 is merged
151+ docker run --rm -v ./models:/models --gpus all -l TensorRT-engines $AI_RUNNER_COMFYUI_IMAGE \
152+ bash -c " cd /workspace/comfystream/src/comfystream/scripts && \
153+ curl -O https://raw.githubusercontent.com/yondonfu/comfystream/535c71a6f665bc1169d07cddd4e2b3cf4edd5a82/src/comfystream/scripts/build_trt.py
154+ python ./build_trt.py \
155+ --model /workspace/ComfyUI/models/unet/dreamshaper-8-dmd-1kstep.safetensors \
156+ --out-engine /workspace/ComfyUI/output/tensorrt/dynamic-dreamshaper8_SD15_\$ dyn-b-1-4-2-h-448-704-512-w-448-704-512_00001_.engine \
157+ --width 512 \
158+ --height 512 \
159+ --min-width 448 \
160+ --min-height 448 \
161+ --max-width 704 \
162+ --max-height 704 \
163+ adduser $( id -u -n) && \
164+ chown -R $( id -u -n) :$( id -g -n) /models" ||
165+ (
166+ echo " failed ComfyUI build_trt.py dynamic engine"
167+ return 1
168+ )
147169}
148170
149171# Download models with a restrictive license.
Original file line number Diff line number Diff line change 1- ARG BASE_IMAGE=livepeer/comfyui-base@sha256:35c82613ba878ba984ba64fdff374d2e51f1815971e5100bfd214d8156669922
1+ ARG BASE_IMAGE=livepeer/comfyui-base@sha256:4435bad85c3a2fce2b491135bee49eedb8edbd8bdf5d124cb0a95a1d4ecb6856
22FROM ${BASE_IMAGE}
33
44# -----------------------------------------------------------------------------
You can’t perform that action at this time.
0 commit comments