Load model without the image size

The current version of TF Stable Diffusion needs the image width and height to load the model.

`def get_models(img_height, img_width, download_weights=True):`

Indeed, the graph is built again if there is another image with a different size.

Is it possible to load the model without specifying the width and height? It would save a lot of time...