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