Skip to content

Commit 56c28fa

Browse files
authored
feat: build static engines for portrait and landscape resolution (yondonfu#211)
* update resolution for 16:9 aspect ratio * add static engines for portrait and landscape resolution
1 parent 0c3cff5 commit 56c28fa

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

docker/entrypoint.sh

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -49,18 +49,24 @@ if [ "$1" = "--build-engines" ]; then
4949
cd /workspace/comfystream
5050
conda activate comfystream
5151

52-
# Build Static Engine for Dreamshaper
53-
python src/comfystream/scripts/build_trt.py --model /workspace/ComfyUI/models/unet/dreamshaper-8-dmd-1kstep.safetensors --out-engine /workspace/ComfyUI/output/tensorrt/static-dreamshaper8_SD15_\$stat-b-1-h-512-w-512_00001_.engine
52+
# Build Static Engine for Dreamshaper - Square (512x512)
53+
python src/comfystream/scripts/build_trt.py --model /workspace/ComfyUI/models/unet/dreamshaper-8-dmd-1kstep.safetensors --out-engine /workspace/ComfyUI/output/tensorrt/static-dreamshaper8_SD15_\$stat-b-1-h-512-w-512_00001_.engine --width 512 --height 512
54+
55+
# Build Static Engine for Dreamshaper - Portrait (384x704)
56+
python src/comfystream/scripts/build_trt.py --model /workspace/ComfyUI/models/unet/dreamshaper-8-dmd-1kstep.safetensors --out-engine /workspace/ComfyUI/output/tensorrt/static-dreamshaper8_SD15_\$stat-b-1-h-384-w-704_00001_.engine --width 384 --height 704
57+
58+
# Build Static Engine for Dreamshaper - Landscape (704x384)
59+
python src/comfystream/scripts/build_trt.py --model /workspace/ComfyUI/models/unet/dreamshaper-8-dmd-1kstep.safetensors --out-engine /workspace/ComfyUI/output/tensorrt/static-dreamshaper8_SD15_\$stat-b-1-h-704-w-384_00001_.engine --width 704 --height 384
5460

5561
# Build Dynamic Engine for Dreamshaper
5662
python src/comfystream/scripts/build_trt.py \
5763
--model /workspace/ComfyUI/models/unet/dreamshaper-8-dmd-1kstep.safetensors \
58-
--out-engine /workspace/ComfyUI/output/tensorrt/dynamic-dreamshaper8_SD15_\$dyn-b-1-4-2-h-448-704-512-w-448-704-512_00001_.engine \
59-
--width 512 \
60-
--height 512 \
61-
--min-width 448 \
62-
--min-height 448 \
63-
--max-width 704 \
64+
--out-engine /workspace/ComfyUI/output/tensorrt/dynamic-dreamshaper8_SD15_\$dyn-b-1-4-2-h-512-704-w-320-384-448_00001_.engine \
65+
--width 384 \
66+
--height 704 \
67+
--min-width 320 \
68+
--min-height 512 \
69+
--max-width 448 \
6470
--max-height 704
6571

6672
# Build Engine for Depth Anything V2

0 commit comments

Comments
 (0)