stabilityai

/

stable-diffusion-xl-1.0-tensorrt

like 154

Follow

Stability AI 36.8k

For dynamic shaping, I use the --build-dynamic-shape flag. But it outputs very small images. For example I give 512x768 for size, but it throws 64x96. How can I compile the model for dynamic shaping? My current command is look like:

"""
python3 code.py "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" --num-warmup-runs 1 --width 768 --height 512 --denoising-steps 25 --onnx-base-dir /workspace/stable-diffusion-xl-1.0-tensorrt/sdxl-1.0-base --repeat-prompt 2 --build-dynamic-shape
"""