AUTOMATIC1111 / stable-diffusion-webui-tensorrt

MIT License
310 stars 20 forks source link

Added Batch Processing, Shape appending, Dynamic alert for too large of a shape for converting trt models. #44

Open jebarpg opened 1 year ago

jebarpg commented 1 year ago

Added the ability to batch process a directory of .safetensor and .ckpt models. Added the ability to append the shape of the converted trt model to the model name. i.e. my_model_1x512x512.trt 1 batch size, max width 512, max height 512. To help differentiate between different trt model conversion shapes for the same base model. Added dynamic alert at the top for when you have gone over the limits of the shape a trt model can be generated at as you slide the max width, height, and batch size sliders.