Skip to content

Commit d7ad557

Browse files
authored
Remove incorrect script download & fix command (#534)
1 parent 3389639 commit d7ad557

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

runner/dl_checkpoints.sh

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -146,20 +146,17 @@ function build_tensorrt_models() {
146146
)
147147

148148
# 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
151149
docker run --rm -v ./models:/models --gpus all -l TensorRT-engines $AI_RUNNER_COMFYUI_IMAGE \
152150
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 \
151+
$CONDA_PYTHON ./build_trt.py \
155152
--model /workspace/ComfyUI/models/unet/dreamshaper-8-dmd-1kstep.safetensors \
156153
--out-engine /workspace/ComfyUI/output/tensorrt/dynamic-dreamshaper8_SD15_\$dyn-b-1-4-2-h-448-704-512-w-448-704-512_00001_.engine \
157154
--width 512 \
158155
--height 512 \
159156
--min-width 448 \
160157
--min-height 448 \
161158
--max-width 704 \
162-
--max-height 704 \
159+
--max-height 704 && \
163160
adduser $(id -u -n) && \
164161
chown -R $(id -u -n):$(id -g -n) /models" ||
165162
(

0 commit comments

Comments
 (0)