Skip to content

Commit 3389639

Browse files
pschroedleliteprox
andauthored
add build of dynamic trt dreamshaper engine (#522)
* add build of dynamic trt dreamshaper engine * bump comfyui-base for multiresolution --------- Co-authored-by: Elite <john@eliteencoder.net>
1 parent 6a59ca1 commit 3389639

File tree

2 files changed

+23
-1
lines changed

2 files changed

+23
-1
lines changed

runner/dl_checkpoints.sh

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff 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.

runner/docker/Dockerfile.live-base-comfyui

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
ARG BASE_IMAGE=livepeer/comfyui-base@sha256:35c82613ba878ba984ba64fdff374d2e51f1815971e5100bfd214d8156669922
1+
ARG BASE_IMAGE=livepeer/comfyui-base@sha256:4435bad85c3a2fce2b491135bee49eedb8edbd8bdf5d124cb0a95a1d4ecb6856
22
FROM ${BASE_IMAGE}
33

44
# -----------------------------------------------------------------------------

0 commit comments

Comments
 (0)