Skip to content

Commit 2097ee0

Browse files
authored
fix check for existing streamdiffusion engines (yondonfu#219)
1 parent 71fb2d7 commit 2097ee0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docker/entrypoint.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ if [ "$1" = "--build-engines" ]; then
9898
fi
9999

100100
# Build Engine for StreamDiffusion
101-
if [ ! -f "$TENSORRT_DIR/StreamDiffusion-engines/stream_diffusion_v2_1_fp16.engine" ]; then #TODO: fix relevant file to check
101+
if [ ! -f "$TENSORRT_DIR/StreamDiffusion-engines/stabilityai/sd-turbo--lcm_lora-True--tiny_vae-True--max_batch-3--min_batch-3--mode-img2img/unet.engine.opt.onnx" ]; then
102102
cd /workspace/ComfyUI/custom_nodes/ComfyUI-StreamDiffusion
103103
MODELS="stabilityai/sd-turbo KBlueLeaf/kohaku-v2.1"
104104
TIMESTEPS="3"

0 commit comments

Comments
 (0)