File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments